Update app.py

app.py

@@ -225,6 +225,7 @@ def analyze_single_message(text, thresholds):
     sentiment = "supportive" if result['label'] == "POSITIVE" else "undermining"
     sentiment_score = result['score'] if sentiment == "undermining" else 0.0
     weapon_flag = detect_weapon_language(text)
+
     adjusted_thresholds = {
         k: v + 0.05 if sentiment == "supportive" else v
         for k, v in thresholds.items()
@@ -240,7 +241,7 @@ def analyze_single_message(text, thresholds):
     threshold_labels = [
         label for label, score in zip(LABELS, scores)
         if score > adjusted_thresholds[label]
-]
+    ]
 
     motifs = [phrase for _, phrase in matched_phrases]
 
@@ -250,43 +251,38 @@ def analyze_single_message(text, thresholds):
         sentiment_after=sentiment_score,
         motifs_found=motifs,
         contradiction_flag=contradiction_flag
-)
+    )
+
+    # Top 2 patterns for display (raw scores)
     top_patterns = sorted(
         [(label, score) for label, score in zip(LABELS, scores)],
         key=lambda x: x[1],
         reverse=True
     )[:2]
 
+    # ✅ Final abuse score: only use patterns that passed threshold
     matched_scores = [
-    (label, score, PATTERN_WEIGHTS.get(label, 1.0))
-    for label, score in zip(LABELS, scores)
-    if score > adjusted_thresholds[label]
-]
+        (label, score, PATTERN_WEIGHTS.get(label, 1.0))
+        for label, score in zip(LABELS, scores)
+        if score > adjusted_thresholds[label]
+    ]
+
     if matched_scores:
         weighted_total = sum(score * weight for _, score, weight in matched_scores)
         weight_sum = sum(weight for _, _, weight in matched_scores)
         abuse_score_raw = (weighted_total / weight_sum) * 100
     else:
         abuse_score_raw = 0
-    print(f"Matched patterns used for abuse scoring: {[(l, round(s, 3)) for l, s, _ in matched_scores]}")
-    print(f"Abuse Score Raw: {round(abuse_score_raw, 1)}")
-    if threshold_labels:
-        stage = get_risk_stage(threshold_labels, sentiment)
-    else:
-        stage = 1  # default to tension-building if no patterns matched
-    if weapon_flag:
-        abuse_score_raw = min(abuse_score_raw + 25, 100)  # boost intensity
+
+    # ✅ Always assign stage
+    stage = get_risk_stage(threshold_labels, sentiment) if threshold_labels else 1
     if weapon_flag and stage < 2:
         stage = 2
-    if weapon_flag:
-        print("⚠️ Weapon-related language detected.")
 
-    if "threat" in threshold_labels or "control" in threshold_labels or "insults" in threshold_labels:
-        abuse_score = min(abuse_score_raw, 100)
-    else:
-        abuse_score = min(abuse_score_raw, 95)
+    if weapon_flag:
+        abuse_score_raw = min(abuse_score_raw + 25, 100)
 
-    
+    abuse_score = min(abuse_score_raw, 100 if "threat" in threshold_labels or "control" in threshold_labels else 95)
 
     print("\n--- Debug Info ---")
     print(f"Text: {text}")
@@ -295,6 +291,8 @@ def analyze_single_message(text, thresholds):
     for label, score in zip(LABELS, scores):
         passed = "✅" if score > adjusted_thresholds[label] else "❌"
         print(f"  {label:25} → {score:.3f} {passed}")
+    print(f"Matched for score: {[(l, round(s, 3)) for l, s, _ in matched_scores]}")
+    print(f"Abuse Score Raw: {round(abuse_score_raw, 1)}")
     print(f"Motifs: {motifs}")
     print(f"Contradiction: {contradiction_flag}")
     print("------------------\n")