Update app.py

app.py

@@ -139,9 +139,9 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
         risk_level = "low"
 
     out = f"Abuse Intensity: {composite_abuse}%\n"
-    out += f"Escalation Potential: {risk_level.capitalize()} ({escalation_score}/{sum(w for _, w in ESCALATION_QUESTIONS)})\n"
-    out += f"Top Pattern: {top_label}\n"
-    out += f"\n{stage_text}"
+    out += f"Escalation Potential: {risk_level.capitalize()} ({escalation_score}/{sum(w for _, w in ESCALATION_QUESTIONS)})"
+    out += generate_risk_snippet(composite_abuse, top_label, escalation_score)
+    out += f"\n\n{stage_text}"
     return out
 
 textbox_inputs = [gr.Textbox(label=f"Message {i+1}") for i in range(3)]