Update app.py

app.py

@@ -277,7 +277,10 @@ def analyze_single_message(text, thresholds):
         abuse_score_raw = (weighted_total / weight_sum) * 100
     else:
         abuse_score_raw = 0
+    if threshold_labels:
         stage = get_risk_stage(threshold_labels, sentiment)
+    else:
+        stage = 1  # default to tension-building if no patterns matched
     if weapon_flag:
         abuse_score_raw = min(abuse_score_raw + 25, 100)  # boost intensity
     if weapon_flag and stage < 2:
@@ -328,6 +331,8 @@ def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
         return "Please enter at least one message."
 
     results = [(analyze_single_message(m, THRESHOLDS.copy()), d) for m, d in active]
+    for result, date in results:
+        assert len(result) == 6, "Unexpected output from analyze_single_message"
     abuse_scores = [r[0][0] for r in results]
     top_labels = [r[0][1][0] if r[0][1] else r[0][2][0][0] for r in results]
     top_scores = [r[0][2][0][1] for r in results]