Update app.py

app.py

@@ -15,30 +15,39 @@ model = RobertaForSequenceClassification.from_pretrained(model_name, trust_remot
 tokenizer = RobertaTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
 LABELS = [
-    "gaslighting", "mockery", "dismissiveness", "control", "guilt_tripping", "apology_baiting", "blame_shifting", "projection",
-    "contradictory_statements", "manipulation", "deflection", "insults", "obscure_formal", "recovery_phase"
+    "blame shifting", "contradictory statements", "control", "dismissiveness",
+    "gaslighting", "guilt tripping", "insults", "obscure language",
+    "projection", "recovery phase", "threat"
 ]
 
 THRESHOLDS = {
-    "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
+    "blame shifting": 0.23,
+    "contradictory statements": 0.25,
+    "control": 0.40,
+    "dismissiveness": 0.45,
+    "gaslighting": 0.30,
+    "guilt tripping": 0.20,
+    "insults": 0.34,
+    "obscure language": 0.25,
+    "projection": 0.35,
+    "recovery phase": 0.25,
+    "threat": 0.25
 }
 
+PATTERN_LABELS = LABELS
+
 EXPLANATIONS = {
-    "gaslighting": "Gaslighting involves making someone question their own reality or perceptions...",
-    "blame_shifting": "Blame-shifting is when one person redirects the responsibility...",
-    "projection": "Projection involves accusing the victim of behaviors the abuser exhibits.",
-    "dismissiveness": "Dismissiveness is belittling or disregarding another person’s feelings.",
-    "mockery": "Mockery ridicules someone in a hurtful, humiliating way.",
-    "recovery_phase": "Recovery phase dismisses someone's emotional healing process.",
-    "insults": "Insults are derogatory remarks aimed at degrading someone.",
-    "apology_baiting": "Apology-baiting manipulates victims into apologizing for abuser's behavior.",
-    "deflection": "Deflection avoids accountability by redirecting blame.",
-    "control": "Control restricts autonomy through manipulation or coercion.",
-    "guilt_tripping": "Guilt-tripping uses guilt to manipulate someone’s actions.",
-    "manipulation": "Manipulation deceives to influence or control outcomes.",
-    "obscure_formal": "Obscure/formal language manipulates through confusion or superiority."
+    "blame shifting": "Blame-shifting is when one person redirects responsibility onto someone else to avoid accountability.",
+    "contradictory statements": "Contradictory statements confuse the listener by flipping positions or denying previous claims.",
+    "control": "Control restricts another person’s autonomy through coercion, manipulation, or threats.",
+    "dismissiveness": "Dismissiveness is belittling or disregarding another person’s feelings, needs, or opinions.",
+    "gaslighting": "Gaslighting involves making someone question their own reality, memory, or perceptions.",
+    "guilt tripping": "Guilt-tripping uses guilt to manipulate someone’s actions or decisions.",
+    "insults": "Insults are derogatory or demeaning remarks meant to shame, belittle, or hurt someone.",
+    "obscure language": "Obscure language manipulates through complexity, vagueness, or superiority to confuse the other person.",
+    "projection": "Projection accuses someone else of the very behaviors or intentions the speaker is exhibiting.",
+    "recovery phase": "Recovery phase statements attempt to soothe or reset tension without acknowledging harm or change.",
+    "threat": "Threats use fear of harm (physical, emotional, or relational) to control or intimidate someone."
 }
 
 PATTERN_WEIGHTS = {