Update app.py

app.py

@@ -23,7 +23,26 @@ TOTAL_LABELS = 17
 # - First 14 are abuse pattern categories
 # - Last 3 are Danger Assessment cues
 TOTAL_LABELS = 17
-
+# Individual thresholds for each of the 17 labels
+THRESHOLDS = {
+    "gaslighting": 0.15,
+    "mockery": 0.15,
+    "dismissiveness": 0.15,
+    "control": 0.15,
+    "guilt_tripping": 0.15,
+    "apology_baiting": 0.15,
+    "blame_shifting": 0.15,
+    "projection": 0.15,
+    "contradictory_statements": 0.15,
+    "manipulation": 0.15,
+    "deflection": 0.15,
+    "insults": 0.15,
+    "obscure_formal": 0.15,
+    "recovery_phase": 0.15,
+    "suicidal_threat": 0.10,
+    "physical_threat": 0.10,
+    "extreme_control": 0.10
+}
 def analyze_messages(input_text):
     input_text = input_text.strip()
     if not input_text:
@@ -45,13 +64,15 @@ def analyze_messages(input_text):
     # Debug printing (remove once you're confident everything works)
     print("Scores:", scores)
 
-    # First 14 = pattern scores
-    pattern_scores = scores[:14]
-    pattern_count = int(np.sum(pattern_scores > 0.15))
+    pattern_count = 0
+    danger_flag_count = 0
 
-    # Last 3 = danger cues
-    danger_scores = scores[14:]
-    danger_flag_count = int(np.sum(danger_scores > 0.20))
+    for i, (label, score) in enumerate(zip(LABELS, scores)):
+    if score > THRESHOLDS[label]:
+        if i < 14:
+            pattern_count += 1
+        else:
+            danger_flag_count += 1
     # (Optional) Print label-by-label for debugging
     for i, s in enumerate(scores):
         print(LABELS[i], "=", round(s, 3))