Update app.py

app.py

@@ -67,47 +67,51 @@ except Exception as e:
 
 style_prompts = {
     'drippy': [
-        "avant-garde streetwear",
-        "high-fashion designer outfit",
-        "trendsetting urban attire",
-        "luxury sneakers and chic accessories",
-        "cutting-edge, bold style",
-        "tailored suit with designer shoes",
-        "modern formalwear with flair",
-        "sleek all-black techwear fit",
-        "runway-ready monochrome look",
-        "statement outerwear with luxury detailing",
-        "elevated business casual with premium materials",
-        "fashion-forward layering with color coordination",
-        "comfortable yet stylish look"
+        "High-fashion runway look with designer labels",
+        "Coordinated color palette with premium fabrics",
+        "Trend-forward streetwear with luxury accessories",
+        "Tailored silhouettes with intentional layering",
+        "Seasonal trend elements executed flawlessly",
+        "Cohesive outfit with statement pieces",
+        "Celebrity red carpet-level styling",
+        "Bold pattern mixing that works harmoniously"
     ],
     'mid': [
-        "casual everyday outfit",
-        "modern minimalistic attire",
-        "simple, relaxed streetwear",
-        "balanced, practical fashion",
-        "smart casual outfit with neutral tones",
-        "clean hoodie with jeans",
-        "basic tee and joggers",
-        "store-bought coordinated outfit",
-        "mid-range branded streetwear",
-        "plain bomber jacket and fitted pants"
+        "Basic wardrobe staples without flair",
+        "Safe color combinations with minimal accessories",
+        "Mass-market fast fashion pieces",
+        "Functional over fashionable aesthetic",
+        "Trend-adjacent but poorly executed",
+        "Mismatched proportions in clothing",
+        "Overly casual for the occasion",
+        "Decade-outdated trend revival"
     ],
     'not_drippy': [
-        "disheveled outfit",
-        "poorly coordinated fashion",
-        "unfashionable, outdated attire",
-        "tacky, mismatched ensemble",
-        "sloppy, uninspired look",
-        "wrinkled clothes with clashing colors",
-        "oversized tee and sagging pants with slides",
-        "random logos and no fit cohesion",
-        "stained or ill-fitting garments",
-        "childish graphic tee with athletic shorts",
-        "pajama bottoms worn outside",
-        "shoes completely unrelated to outfit"
+        "Severely mismatched color schemes",
+        "Pilled/stained/faded clothing items",
+        "Ill-fitting garments in multiple pieces",
+        "Clashing patterns without cohesion",
+        "Inappropriate footwear for outfit context",
+        "Overly literal costume-like styling",
+        "Multiple competing trends in one outfit",
+        "Wrinkled/unkempt fabric presentation"
     ]
 }
+style_prompts['drippy'] += [
+    "coordinated color scheme",
+    "perfect garment proportions", 
+    "current fashion trends",
+    "high-quality materials",
+    "complementary accessories"
+]
+
+style_prompts['not_drippy'] += [
+    "clashing colors",
+    "poor fit proportions",
+    "outdated trends",
+    "cheap-looking fabrics",
+    "missing accessories"
+]
 
 
 clothing_prompts = [
@@ -297,7 +301,7 @@ def analyze_outfit(input_img):
 
         # Determine overall style category AND DEFINE score_label
         score_label = "Style Score" # Initialize with a default/fallback
-        if drip_score > 0.37 and drip_score > mid_score and drip_score > not_score:
+        """if drip_score > 0.35 and drip_score > mid_score and drip_score > not_score:
             category_key = 'drippy'
             final_score = drip_score
             score_label = "Drip Score" 
@@ -313,7 +317,23 @@ def analyze_outfit(input_img):
         category_label = CATEGORY_LABEL_MAP[category_key]
        
         percentage_score = max(0, final_score * 100)
-        percentage_score_str = f"{percentage_score:.0f}%" 
+        percentage_score_str = f"{percentage_score:.0f}%""""
+
+        final_score = max(drip_score, mid_score, not_score)
+        percentage_score = final_score * 100
+        
+        if percentage_score >= 75:
+            category_key = 'drippy'
+            score_label = "Drip Score"
+        elif percentage_score >= 40:
+            category_key = 'mid'
+            score_label = "Mid Score"
+        else:
+            category_key = 'not_drippy'
+            score_label = "Trash Score"
+        
+        category_label = CATEGORY_LABEL_MAP[category_key]
+        percentage_score_str = f"{percentage_score:.0f}%"
 
       
         print(f"Style analysis: Category={category_label}, Score = {score_label}={percentage_score_str} (Raw Score: {final_score:.4f})")