Update app.py

app.py

@@ -248,9 +248,16 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
     responses_checked = any(answers_and_none[:-1])
     none_selected = not responses_checked and none_selected_checked
 
-    escalation_score = None if none_selected else sum(
-        w for (response, (q, w)) in zip(answers_and_none[:-1], ESCALATION_QUESTIONS) if response
-    )
+    if none_selected:
+        escalation_score = None
+        risk_level = "unknown"
+    else:
+        escalation_score = sum(w for (_, w), a in zip(ESCALATION_QUESTIONS, answers_and_none[:-1]) if a)
+        risk_level = (
+            "High" if escalation_score >= 16 else
+            "Moderate" if escalation_score >= 8 else
+            "Low"
+        )
 
     messages = [msg1, msg2, msg3]
     active = [m for m in messages if m.strip()]
@@ -267,6 +274,7 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
 
     composite_abuse = int(round(sum(abuse_scores) / len(abuse_scores)))
     top_label = f"{top_labels[0]} – {int(round(top_scores[0] * 100))}%"
+
     most_common_stage = max(set(stages), key=stages.count)
     stage_text = RISK_STAGE_LABELS[most_common_stage]
 
@@ -280,19 +288,13 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
     out += "📊 This reflects the strength and severity of detected abuse patterns in the message(s).\n\n"
 
     if escalation_score is None:
-        risk_level = "unknown"
         out += "Escalation Potential: Unknown (Checklist not completed)\n"
         out += "🔍 *This section was not completed. Escalation potential is unknown.*\n\n"
     else:
-        risk_level = (
-            "High" if escalation_score >= 16 else
-            "Moderate" if escalation_score >= 8 else
-            "Low"
-        )
         out += f"Escalation Potential: {risk_level} ({escalation_score}/{sum(w for _, w in ESCALATION_QUESTIONS)})\n"
         out += "🚨 This indicates how many serious risk factors are present based on your answers to the safety checklist.\n\n"
 
-    out += generate_risk_snippet(composite_abuse, top_label, escalation_score or 0, most_common_stage)
+    out += generate_risk_snippet(composite_abuse, top_label, escalation_score if escalation_score is not None else 0, most_common_stage)
     out += f"\n\n{stage_text}"
     out += darvo_blurb