Update app.py

app.py

@@ -401,7 +401,9 @@ def analyze_single_message(text, thresholds):
         "recovery phase", "obscure language"
     ]
 
-    label_scores = {label: score for label, score in zip(LABELS, scores)}
+    # Use top_label from earlier safely, and convert to score if available
+    label_key = top_label.split(" = ")[0]  # Extract raw label (e.g., "Control" from "Control = 78%")
+    score = label_scores.get(label_key)
     label = score_to_label(score) if score is not None else "Unknown"
 
     # 🛡️ Prevent obscure language from being chosen unless it crosses a hard threshold