Update app.py

app.py

@@ -22,10 +22,10 @@ LABELS = [
 ]
 
 THRESHOLDS = {
-    "gaslighting": 0.25, "mockery": 0.15, "dismissiveness": 0.30, "control": 0.43, "guilt_tripping": 0.19,
-    "apology_baiting": 0.45, "blame_shifting": 0.23, "projection": 0.50, "contradictory_statements": 0.25,
+    "gaslighting": 0.25, "mockery": 0.15, "dismissiveness": 0.45, "control": 0.43, "guilt_tripping": 0.15,
+    "apology_baiting": 0.2, "blame_shifting": 0.23, "projection": 0.50, "contradictory_statements": 0.25,
     "manipulation": 0.25, "deflection": 0.30, "insults": 0.34, "obscure_formal": 0.25, "recovery_phase": 0.25,
-    "non_abusive": 2.0, "suicidal_threat": 0.45, "physical_threat": 0.02, "extreme_control": 0.36
+    "non_abusive": 2.0, "suicidal_threat": 0.45, "physical_threat": 0.02, "extreme_control": 0.30
 }
 
 PATTERN_LABELS = LABELS[:15]
@@ -63,11 +63,23 @@ def custom_sentiment(text):
     score = probs[0][label_idx].item()
     return {"label": label, "score": score}
 
+PATTERN_WEIGHTS = {
+    "physical_threat": 1.5,
+    "suicidal_threat": 1.4,
+    "extreme_control": 1.5,
+    "gaslighting": 1.3,
+    "control": 1.2,
+    "dismissiveness": 0.8,
+    "non_abusive": 0.0  # shouldn't contribute to abuse score
+}
+
 def calculate_abuse_level(scores, thresholds, motif_hits=None):
-    triggered_scores = [
-        score for label, score in zip(LABELS, scores) if score > thresholds[label]
-    ]
-    base_score = round(np.mean(triggered_scores) * 100, 2) if triggered_scores else 0.0
+    weighted_scores = []
+    for label, score in zip(LABELS, scores):
+        if score > thresholds[label]:
+            weight = PATTERN_WEIGHTS.get(label, 1.0)
+            weighted_scores.append(score * weight)
+    base_score = round(np.mean(weighted_scores) * 100, 2) if weighted_scores else 0.0
 
     motif_hits = motif_hits or []
     if any(label in motif_hits for label in {"physical_threat", "suicidal_threat", "extreme_control"}):