Update app.py

app.py

@@ -121,7 +121,15 @@ def analyze_messages(input_text, risk_flags):
     scored_patterns = [
         (label, score) for label, score in zip(PATTERN_LABELS, scores[:15]) if label != "non_abusive"
     ]
-    top_patterns = sorted(scored_patterns, key=lambda x: x[1], reverse=True)[:2]
+    # Override top patterns if a high-risk motif was detected
+    override_labels = {"physical_threat", "suicidal_threat", "extreme_control"}
+    override_matches = [label for label, _ in matched_phrases if label in override_labels]
+
+    if override_matches:
+        top_patterns = [(label, 1.0) for label in override_matches]
+    else:
+        top_patterns = sorted(scored_patterns, key=lambda x: x[1], reverse=True)[:2]
+        top_patterns = sorted(scored_patterns, key=lambda x: x[1], reverse=True)[:2]
 
     top_pattern_explanations = "\n".join([
         f"• {label.replace('_', ' ').title()}: {EXPLANATIONS.get(label, 'No explanation available.')}"