Update app.py

app.py

@@ -347,20 +347,31 @@ def get_risk_stage(patterns, sentiment):
         return 4
     return 1
 
-def generate_risk_snippet(abuse_score, top_label, escalation_score, stage):
+def generate_risk_snippet(abuse_score, top_label, escalation_score, stage, pattern_scores):
+    aggression_score = pattern_scores.get("aggression", 0)
+
+    # Updated risk level logic
     if abuse_score >= 85 or escalation_score >= 16:
         risk_level = "high"
-    elif abuse_score >= 60 or escalation_score >= 8:
+    elif abuse_score >= 60 or escalation_score >= 8 or aggression_score >= 0.75:
         risk_level = "moderate"
     elif stage == 2 and abuse_score >= 40:
-        risk_level = "moderate"  # 🔧 New rule for escalation stage
+        risk_level = "moderate"
     else:
         risk_level = "low"
-    if isinstance(top_label, str) and " – " in top_label:
-        pattern_label, pattern_score = top_label.split(" – ")
+
+    # Narrative
+    base = f"\n\n🛑 Risk Level: {risk_level.capitalize()}\n"
+    base += f"This message shows strong indicators of **{top_label}**. "
+
+    if risk_level == "high":
+        base += "The language may reflect patterns of emotional control, even when expressed in soft or caring terms.\n"
+    elif risk_level == "moderate":
+        base += "There are signs of emotional pressure or verbal aggression that may escalate if repeated.\n"
     else:
-        pattern_label = str(top_label) if top_label is not None else "Unknown"
-        pattern_score = ""
+        base += "The message does not strongly indicate abuse, but it's important to monitor for patterns.\n"
+
+    return risk_level, base
 
     WHY_FLAGGED = {
         "control": "This message may reflect efforts to restrict someone’s autonomy, even if it's framed as concern or care.",