Update app.py

app.py

@@ -1,29 +1,29 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionXLPipeline
 
-# Load the Stable Diffusion XL model
+# Load SDXL model
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
-pipe = StableDiffusionPipeline.from_pretrained(
+pipe = StableDiffusionXLPipeline.from_pretrained(
     model_id,
     torch_dtype=torch.float16,
+    safety_checker=None  # Disable safety checker for hackathon demo
 )
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Function to generate outfit image + suggestion with error handling
+# Function to generate outfit image + suggestion
 def generate_outfit(weather, activity, style):
     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"
+            f"{activity} activity, {style} style, front view, 30-degree angle, highly detailed, full body"
         )
-        
-        # Generate image
+        # SDXL generates output as a list of PIL images
         result = pipe(prompt, guidance_scale=7.5)
-        image = result.images[0]  # Ensure we take the first PIL image
+        image = result.images[0]  # take the first image
 
-        # Simple outfit suggestion text
+        # Outfit suggestion
         suggestion_text = (
             f"💡 Suggested outfit for {weather} weather, {activity} activity, {style} style:\n"
             "- Top: light breathable shirt or jacket depending on weather\n"
@@ -32,39 +32,18 @@ def generate_outfit(weather, activity, style):
             "- 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)}"
 
 # Gradio 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);
-}
+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;}
+.gr-button:hover {background-color:#007f70 !important;}
+.gr-image {border-radius:12px; box-shadow:0 0 20px rgba(0,255,255,0.1);}
 """
 
 with gr.Blocks(css=custom_css) as app:
@@ -73,15 +52,29 @@ with gr.Blocks(css=custom_css) as app:
 
     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)
 
-    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]
+    )
 
 app.launch(share=True)