Update app.py

app.py

@@ -16,42 +16,20 @@ clip_model, clip_preprocess = clip.load("ViT-B/32", device=device)
 yolo_model = YOLO('yolov8n.pt').to(device)
 fashion_model = YOLO('best.pt').to(device)
 
-# Style prompts and categories
 style_prompts = {
-    'drippy': [
-        "avant-garde streetwear", "high-fashion designer outfit", "trendsetting urban attire",
-        "luxury sneakers and chic accessories", "cutting-edge, bold style"
-    ],
-    'mid': [
-        "casual everyday outfit", "modern minimalistic attire", "comfortable yet stylish look",
-        "simple, relaxed streetwear", "balanced, practical fashion"
-    ],
-    'not_drippy': [
-        "disheveled outfit", "poorly coordinated fashion", "unfashionable, outdated attire",
-        "tacky, mismatched ensemble", "sloppy, uninspired look"
-    ]
+    'drippy': ["avant-garde streetwear", "high-fashion designer outfit", "trendsetting urban attire", "luxury sneakers and chic accessories", "cutting-edge, bold style"],
+    'mid': ["casual everyday outfit", "modern minimalistic attire", "comfortable yet stylish look", "simple, relaxed streetwear", "balanced, practical fashion"],
+    'not_drippy': ["disheveled outfit", "poorly coordinated fashion", "unfashionable, outdated attire", "tacky, mismatched ensemble", "sloppy, uninspired look"]
 }
 
 clothing_prompts = [
-    "t-shirt", "dress shirt", "blouse", "hoodie", "jacket", "sweater", "coat", "dress", "skirt",
-    "pants", "jeans", "trousers", "shorts", "sneakers", "boots", "heels", "sandals", "cap", "hat",
-    "scarf", "gloves", "bag", "accessory", "tank-top", "haircut"
+    "t-shirt", "dress shirt", "blouse", "hoodie", "jacket", "sweater", "coat", "dress", "skirt", "pants", "jeans", "trousers", "shorts", "sneakers", "boots", "heels", "sandals", "cap", "hat", "scarf", "gloves", "bag", "accessory", "tank-top", "haircut"
 ]
 
 response_templates = {
-    'drippy': [
-        "You're Drippy, bruh – fire {item}!", "{item} goes crazy, on god!",
-        "Certified drippy with that {item}."
-    ],
-    'mid': [
-        "Drop the {item} and you might get a text back.", "It's alright, but I'd upgrade the {item}.",
-        "Mid fit alert. That {item} is holding you back."
-    ],
-    'not_drippy': [
-        "Bro thought that {item} was tuff!", "Oh hell nah! Burn that {item}!",
-        "Crimes against fashion, especially that {item}! Also… maybe get a haircut.",
-        "Never walk out the house again with that {item}."
-    ]
+    'drippy': ["You're Drippy, bruh – fire {item}!", "{item} goes crazy, on god!", "Certified drippy with that {item}."],
+    'mid': ["Drop the {item} and you might get a text back.", "It's alright, but I'd upgrade the {item}.", "Mid fit alert. That {item} is holding you back."],
+    'not_drippy': ["Bro thought that {item} was tuff!", "Oh hell nah! Burn that {item}!", "Crimes against fashion, especially that {item}! Also… maybe get a haircut.", "Never walk out the house again with that {item}."]
 }
 
 CATEGORY_LABEL_MAP = {"drippy": "drippy", "mid": "mid", "not_drippy": "trash"}
@@ -84,7 +62,7 @@ def analyze_outfit(img: Image.Image):
     drip_len = len(style_prompts['drippy'])
     mid_len = len(style_prompts['mid'])
     not_len = len(style_prompts['not_drippy'])
-    
+
     drip_score = np.mean(probs[:drip_len])
     mid_score = np.mean(probs[drip_len:drip_len + mid_len])
     not_score = np.mean(probs[drip_len + mid_len:drip_len + mid_len + not_len])
@@ -117,38 +95,29 @@ def analyze_outfit(img: Image.Image):
 
     return category_html, tts_path, response
 
-# Gradio UI layout
+# Gradio Interface
 with gr.Blocks(css="""
 .container { max-width: 600px; margin: 0 auto; padding: 2rem; }
 button { background-color: #1f04ff; color: white; border: none; padding: 0.75rem 1.5rem; border-radius: 6px; cursor: pointer; font-weight: bold; }
 button:hover { background-color: #1500cc; }
 #resultbox { border: 1px solid #e3e3e3; border-radius: 10px; padding: 1rem; background: #fafafa; }
 """) as demo:
-
     with gr.Group(elem_classes=["container"]):
-        input_image = gr.Image(
-            type='pil',
-            label="Upload or Take a Picture",
-            sources=["upload", "webcam", "clipboard"],
-            image_mode="RGB",
-            elem_id="custom-image-input"
-        )
-        gr.Markdown("""
-        <p style='margin-top:-10px; font-size: 0.9rem; color: #555;'>
-        Click the 📷 camera icon to take a photo, the 📁 file icon to upload one, or the 📋 clipboard icon to paste.
-        </p>
-        """)
-        analyze_button = gr.Button("🔥 Analyze My Fit")
+        gr.Markdown("### Upload or Snap a Photo")
+
+        upload_input = gr.Image(type="pil", label="Upload Image", image_mode="RGB")
+        webcam_input = gr.Image(type="pil", source="webcam", label="Take a Photo", image_mode="RGB")
+
+        analyze_button_upload = gr.Button("Analyze Uploaded Fit")
+        analyze_button_webcam = gr.Button("Analyze Webcam Fit")
 
         category_html = gr.HTML()
         audio_output = gr.Audio(autoplay=True, label="")
         response_box = gr.Textbox(label="Response", lines=2, interactive=False)
 
-        analyze_button.click(
-            fn=analyze_outfit,
-            inputs=input_image,
-            outputs=[category_html, audio_output, response_box],
-        )
+        analyze_button_upload.click(fn=analyze_outfit, inputs=upload_input, outputs=[category_html, audio_output, response_box])
+        analyze_button_webcam.click(fn=analyze_outfit, inputs=webcam_input, outputs=[category_html, audio_output, response_box])
 
 if __name__ == '__main__':
     demo.launch()
+