Create app.py

app.py

@@ -0,0 +1,99 @@
+# ClothCast - AI Outfit Forecaster (Hackathon MVP)
+import gradio as gr
+import torch
+from diffusers import StableDiffusionPipeline
+
+# Load Stable Diffusion XL Base model
+model_id = "stabilityai/stable-diffusion-xl-base-1.0"
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id, 
+    torch_dtype=torch.float16
+)
+pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+
+# Function to generate outfit image and suggestion
+def generate_outfit(weather, activity, style):
+    # Build prompt
+    prompt = (
+        f"A realistic full-body outfit for {weather} weather, "
+        f"{activity} activity, {style} style, front view, 30-degree angle, 4k detail"
+    )
+    
+    # Generate image
+    image = pipe(prompt, guidance_scale=7.5).images[0]
+    
+    # Simple outfit suggestion text
+    suggestion_text = (
+        f"💡 Suggested outfit for {weather} weather, {activity} activity, {style} style:\n"
+        "- Top: light breathable shirt or jacket depending on weather\n"
+        "- Bottom: comfortable pants or shorts\n"
+        "- Shoes: suitable for activity\n"
+        "- Accessories: hat, sunglasses, or scarf depending on weather"
+    )
+    return image, suggestion_text
+
+# Custom CSS for polished UI
+custom_css = """
+body {
+    background: linear-gradient(135deg, #232526, #414345);
+    color: white;
+    font-family: 'Poppins', sans-serif;
+}
+.gradio-container {
+    max-width: 900px !important;
+    margin: auto;
+}
+h1, h2, h3 {
+    text-align: center;
+    color: #fff;
+}
+.gr-button {
+    background-color: #00bfa6 !important;
+    color: white !important;
+    font-weight: bold;
+    border-radius: 12px !important;
+    transition: all 0.3s ease-in-out;
+}
+.gr-button:hover {
+    background-color: #007f70 !important;
+}
+.gr-image {
+    border-radius: 12px;
+    box-shadow: 0 0 20px rgba(0,255,255,0.1);
+}
+"""
+
+# Gradio UI
+with gr.Blocks(css=custom_css) as app:
+    gr.Markdown("<h1 style='text-align:center;'>👗 ClothCast - AI Outfit Forecaster</h1>")
+    gr.Markdown("<p style='text-align:center;'>Select your weather, activity, and style to generate an AI outfit!</p>")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            weather = gr.Dropdown(
+                ["Hot", "Cold", "Rainy", "Snowy", "Mild", "Humid"], 
+                label="🌤 Weather", value="Hot"
+            )
+            activity = gr.Dropdown(
+                ["Casual", "Work", "Party", "Sporty"], 
+                label="🏃 Activity", value="Casual"
+            )
+            style = gr.Dropdown(
+                ["Casual", "Sporty", "Formal"], 
+                label="🎨 Style", value="Casual"
+            )
+            generate_btn = gr.Button("Generate Outfit 👗", variant="primary")
+        
+        with gr.Column(scale=1):
+            outfit_image = gr.Image(label="🖼 Generated Outfit")
+            outfit_text = gr.Textbox(label="💡 Outfit Suggestion", lines=8)
+
+    # Connect button to function
+    generate_btn.click(
+        generate_outfit, 
+        inputs=[weather, activity, style], 
+        outputs=[outfit_image, outfit_text]
+    )
+
+# Launch the app
+app.launch(share=True)