Update app.py

app.py

@@ -581,19 +581,11 @@ def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
         risk_level = "unknown"
     else:
         escalation_score = sum(w for (_, w), a in zip(ESCALATION_QUESTIONS, answers_and_none[:-1]) if a)
-
-    # Abuse label escalation logic
-        high = {'Control'}
-        moderate = {
-        'Gaslighting', 'Dismissiveness', 'Obscure Language',
-        'Insults', 'Contradictory Statements', 'Guilt-tripping'
-        }
-        low = {'Blame-shifting', 'Projection', 'Recovery'}
-
-        counts = {'high': 0, 'moderate': 0, 'low': 0}
-    for label in predicted_labels:
-        if label in high:
-            counts['high'] += 1
+        predicted_labels = [label for r in results for label, _ in r[0][2]]
+   
+        for label in predicted_labels:
+            if label in high:
+                counts['high'] += 1
         elif label in moderate:
             counts['moderate'] += 1
         elif label in low: