Update app.py

app.py

@@ -84,28 +84,47 @@ for cat_prompts in style_prompts.values():
     all_prompts.extend(cat_prompts)
 style_prompts_end_index = len(all_prompts) # Mark where style prompts end dont mess up loop
 all_prompts.extend(clothing_prompts)
-print(f"Total prompts for CLIP: {len(all_prompts)}")
+#print(f"Total prompts for CLIP: {len(all_prompts)}")
 
 # Response Templates maybe not call out specific item all the time?
 response_templates = {
     'drippy': [
-        "You're Drippy, bruh – fire {item}!", "{item} goes crazy, on god!", "Certified drippy with that {item}."
+        "You're Drippy, bruh – fire {item}!",
+        "{item} goes crazy, on god!",
+        "Certified drippy with that {item}.",
+        "Your {item} just walked a Paris runway.",
+        "That {item}? Straight from the future.",
+        "You just turned a sidewalk into a runway."
     ],
     '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.", "We can do better come on now", "I dont think you want it enough"
+        "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.",
+        "We can do better come on now",
+        "I don't think you want it enough",
+        "You're teetering on drip... fix the {item}.",
+        "You're in the gray zone. That {item} ain't helping."
     ],
     'not_drippy': [
-        "Bro thought that {item} was tuff!", "Oh hell nah! Burn that {item}!",
+        "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}.",
-        "Your drip is trash, try again.", # Generic trash response
+        "Your drip is trash, try again.",
         "This ain't it chief. The overall style needs work.",
-        "Didn't need an AI to tell you to go back to the wardrobe"
+        "Didn't need an AI to tell you to go back to the wardrobe",
+        "Someone call the fashion police. That {item} needs arresting.",
+        "Your outfit just gave me a 404 error.",
+        "That {item} made my GPU overheat in shame."
     ]
 }
 CATEGORY_LABEL_MAP = { "drippy": "drippy", "mid": "mid", "not_drippy": "trash" }
 
+def format_detected_items(item_list):
+    if not item_list:
+        return ""
+    return "<p class='result-items'>Detected items: " + ", ".join(item_list) + "</p>"
+
 # --- Helper Functions ---
 def get_top_clip_clothing(probs, n=1):
     """Gets the top N clothing items based on CLIP probabilities."""
@@ -216,7 +235,7 @@ def analyze_outfit(input_img: Image.Image):
 
         # Determine overall style category AND DEFINE score_label
         score_label = "Style Score" # Initialize with a default/fallback
-        if drip_score > mid_score and drip_score > not_score:
+        if drip_score > 0.41 and drip_score > mid_score and drip_score > not_score:
             category_key = 'drippy'
             final_score = drip_score
             score_label = "Drip Score" # <<< DEFINE score_label
@@ -287,31 +306,11 @@ def analyze_outfit(input_img: Image.Image):
     # 5) Generate Response and TTS
     try:
         response_pool = response_templates[category_key]
+        # Choose a random template from the entire response pool
+        chosen_template = random.choice(response_pool)
 
-        # If generic response is needed OR category is trash, potentially use more generic templates
-        if generic_response_needed or category_key == 'not_drippy':
-             # Give higher chance to generic trash responses if category is 'not_drippy'
-             if category_key == 'not_drippy':
-                 # Mix specific item templates with generic ones
-                 specific_templates = [t for t in response_pool if '{item}' in t]
-                 generic_templates = [t for t in response_pool if '{item}' not in t]
-                 # e.g., 70% chance generic, 30% chance specific item mention (even if generic item name)
-                 if random.random() < 0.7 or generic_response_needed:
-                     chosen_template = random.choice(generic_templates if generic_templates else response_pool)
-                 else:
-                     chosen_template = random.choice(specific_templates if specific_templates else response_pool)
-             else: # Mid or Drippy, but generic needed
-                 chosen_template = random.choice([t for t in response_pool if '{item}' in t] if not generic_response_needed else response_pool)
-
-        else: # Drippy or Mid, and we have a specific item
-            chosen_template = random.choice([t for t in response_pool if '{item}' in t])
-
-        # Format the response, substituting the determined item name
-        # Handle cases where the chosen template might be generic and doesn't have {item}
-        if '{item}' in chosen_template:
-            response_text = chosen_template.format(item=final_clothing_item)
-        else:
-            response_text = chosen_template # Use the generic template as is
+        # Format the response, substituting the item name if needed
+        response_text = chosen_template.format(item=final_clothing_item) if '{item}' in chosen_template else chosen_template
 
         tts_path = os.path.join(tempfile.gettempdir(), f"drip_{uuid.uuid4().hex}.mp3")
         tts = gTTS(text=response_text, lang='en', tld='com', slow=False)
@@ -325,10 +324,9 @@ def analyze_outfit(input_img: Image.Image):
                 <p class='result-score'>{score_label}: {percentage_score_str}</p>
             </div>
         """
-        #<p class='result-score'>Style Score: {final_score_str}</p>
         return category_html, tts_path, response_text
 
-    except Exception as e:
+        except Exception as e:
         print(f"Error during response/TTS generation: {e}")
         percentage_score = max(0, final_score * 100)
         percentage_score_str = f"{percentage_score:.0f}%"