Update app.py

app.py

@@ -532,34 +532,20 @@ 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)
-        escalation_bump = 0  # Will be computed later
-    hybrid_score = escalation_score + escalation_bump
-    risk_level = (
-        "High" if hybrid_score >= 16 else
-        "Moderate" if hybrid_score >= 8 else
-        "Low"
-)
-
-# Now it's safe to assign these:
-    abuse_scores = [r[0][0] for r in results]
-    stages = [r[0][4] for r in results]
 
-# Post-check override to catch hidden risks
-    if any(score > 70 for score in abuse_scores) or any(stage == 2 for stage in stages):
-        if risk_level == "Low":
-            risk_level = "Moderate"
     messages = [msg1, msg2, msg3]
     dates = [date1, date2, date3]
     active = [(m, d) for m, d in zip(messages, dates) if m.strip()]
     if not active:
         return "Please enter at least one message."
 
+    # Run model on messages
     results = [(analyze_single_message(m, THRESHOLDS.copy()), d) for m, d in active]
-    escalation_bump = 0
 
+    # Calculate escalation bump *after* model results exist
+    escalation_bump = 0
     for result, _ in results:
         abuse_score, threshold_labels, top_patterns, sentiment, stage, darvo_score, tone_tag = result
-
         if darvo_score > 0.65:
             escalation_bump += 3
         if tone_tag in ["forced accountability flip", "emotional threat"]:
@@ -568,12 +554,24 @@ def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
             escalation_bump += 2
         if stage == 2:
             escalation_bump += 3
-    hybrid_score = escalation_score + escalation_bump
+
+    # Now we can safely calculate hybrid_score
+    hybrid_score = escalation_score + escalation_bump if escalation_score is not None else 0
     risk_level = (
         "High" if hybrid_score >= 16 else
         "Moderate" if hybrid_score >= 8 else
         "Low"
     )
+
+    # Now compute scores and allow override
+    abuse_scores = [r[0][0] for r in results]
+    stages = [r[0][4] for r in results]
+
+    # Post-check override (e.g. stage 2 or high abuse score forces Moderate risk)
+    if any(score > 70 for score in abuse_scores) or any(stage == 2 for stage in stages):
+        if risk_level == "Low":
+            risk_level = "Moderate"
+    )
     for result, date in results:
         assert len(result) == 7, "Unexpected output from analyze_single_message"
     abuse_scores = [r[0][0] for r in results]