Update app.py

app.py

@@ -1,38 +1,42 @@
-# ClothCast - AI Outfit Forecaster (Hackathon MVP)
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
 
-# Load Stable Diffusion XL Base model
+# Load the Stable Diffusion XL model
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 pipe = StableDiffusionPipeline.from_pretrained(
-    model_id, 
-    torch_dtype=torch.float16
+    model_id,
+    torch_dtype=torch.float16,
 )
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Function to generate outfit image and suggestion
+# Function to generate outfit image + suggestion with error handling
 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
+    try:
+        # 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
+        result = pipe(prompt, guidance_scale=7.5)
+        image = result.images[0]  # Ensure we take the first PIL image
+
+        # 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
+    except Exception as e:
+        # If something fails, return a placeholder and error text
+        return None, f"❌ Error generating outfit: {str(e)}"
 
-# Custom CSS for polished UI
+# Gradio UI
 custom_css = """
 body {
     background: linear-gradient(135deg, #232526, #414345);
@@ -63,37 +67,21 @@ h1, h2, h3 {
 }
 """
 
-# 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"
-            )
+            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]
-    )
+    generate_btn.click(generate_outfit, inputs=[weather, activity, style], outputs=[outfit_image, outfit_text])
 
-# Launch the app
 app.launch(share=True)
+