Update app.py

app.py

@@ -348,18 +348,22 @@ def get_risk_stage(patterns, sentiment):
     return 1
 
 def generate_risk_snippet(abuse_score, top_label, escalation_score, stage):
-    # Extract aggression score from the top label if it's aggression
+    import re
+
+    # Extract aggression score if aggression is detected
     if isinstance(top_label, str) and "aggression" in top_label.lower():
         try:
-            aggression_score = int(top_label.split("–")[1].replace("%", "").strip()) / 100
+            match = re.search(r"\(?(\d+)\%?\)?", top_label)
+            aggression_score = int(match.group(1)) / 100 if match else 0
         except:
             aggression_score = 0
     else:
         aggression_score = 0
 
+    # Revised risk logic
     if abuse_score >= 85 or escalation_score >= 16:
         risk_level = "high"
-    elif abuse_score >= 60 or escalation_score >= 8 or aggression_score >= 0.75:
+    elif abuse_score >= 60 or escalation_score >= 8 or aggression_score >= 0.25:
         risk_level = "moderate"
     elif stage == 2 and abuse_score >= 40:
         risk_level = "moderate"