Update app.py

app.py

@@ -49,7 +49,41 @@ ESCALATION_QUESTIONS = [
     ("Violence has increased in frequency or severity", 3),
     ("Partner monitors your calls/GPS/social media", 2)
 ]
-
+DARVO_MOTIFS = [
+    "I never said that.", "You’re imagining things.", "That never happened.",
+    "You’re making a big deal out of nothing.", "It was just a joke.", "You’re too sensitive.",
+    "I don’t know what you’re talking about.", "You’re overreacting.", "I didn’t mean it that way.",
+    "You’re twisting my words.", "You’re remembering it wrong.", "You’re always looking for something to complain about.",
+    "You’re just trying to start a fight.", "I was only trying to help.", "You’re making things up.",
+    "You’re blowing this out of proportion.", "You’re being paranoid.", "You’re too emotional.",
+    "You’re always so dramatic.", "You’re just trying to make me look bad.",
+
+    "You’re crazy.", "You’re the one with the problem.", "You’re always so negative.",
+    "You’re just trying to control me.", "You’re the abusive one.", "You’re trying to ruin my life.",
+    "You’re just jealous.", "You’re the one who needs help.", "You’re always playing the victim.",
+    "You’re the one causing all the problems.", "You’re just trying to make me feel guilty.",
+    "You’re the one who can’t let go of the past.", "You’re the one who’s always angry.",
+    "You’re the one who’s always complaining.", "You’re the one who’s always starting arguments.",
+    "You’re the one who’s always making things worse.", "You’re the one who’s always making me feel bad.",
+    "You’re the one who’s always making me look like the bad guy.",
+    "You’re the one who’s always making me feel like a failure.",
+    "You’re the one who’s always making me feel like I’m not good enough.",
+
+    "I can’t believe you’re doing this to me.", "You’re hurting me.",
+    "You’re making me feel like a terrible person.", "You’re always blaming me for everything.",
+    "You’re the one who’s abusive.", "You’re the one who’s controlling.", "You’re the one who’s manipulative.",
+    "You’re the one who’s toxic.", "You’re the one who’s gaslighting me.",
+    "You’re the one who’s always putting me down.", "You’re the one who’s always making me feel bad.",
+    "You’re the one who’s always making me feel like I’m not good enough.",
+    "You’re the one who’s always making me feel like I’m the problem.",
+    "You’re the one who’s always making me feel like I’m the bad guy.",
+    "You’re the one who’s always making me feel like I’m the villain.",
+    "You’re the one who’s always making me feel like I’m the one who needs to change.",
+    "You’re the one who’s always making me feel like I’m the one who’s wrong.",
+    "You’re the one who’s always making me feel like I’m the one who’s crazy.",
+    "You’re the one who’s always making me feel like I’m the one who’s abusive.",
+    "You’re the one who’s always making me feel like I’m the one who’s toxic."
+]
 def detect_contradiction(message):
     patterns = [
         (r"\b(i love you).{0,15}(i hate you|you ruin everything)", re.IGNORECASE),
@@ -60,6 +94,27 @@ def detect_contradiction(message):
         (r"\b(i guess i’m just).{0,15}(the bad guy|worthless|never enough)", re.IGNORECASE)
     ]
     return any(re.search(p, message, flags) for p, flags in patterns)
+    
+def calculate_darvo_score(patterns, sentiment_before, sentiment_after, motifs_found, contradiction_flag=False):
+    pattern_hits = len([p for p in patterns if p in DARVO_PATTERNS])
+    pattern_score = pattern_hits / len(DARVO_PATTERNS)
+    
+    sentiment_shift_score = max(0.0, sentiment_after - sentiment_before)
+
+    motif_hits = len([
+        motif for motif in motifs_found
+        if any(phrase.lower() in motif.lower() for phrase in DARVO_MOTIFS)
+    ])
+    motif_score = motif_hits / len(DARVO_MOTIFS)
+
+    contradiction_score = 1.0 if contradiction_flag else 0.0
+
+    return round(min(
+        0.3 * pattern_score +
+        0.3 * sentiment_shift_score +
+        0.25 * motif_score +
+        0.15 * contradiction_score, 1.0
+    ), 3)
 
 def get_risk_stage(patterns, sentiment):
     if "threat" in patterns or "insults" in patterns:
@@ -126,7 +181,13 @@ def analyze_single_message(text, thresholds):
 
     contradiction_flag = detect_contradiction(text)
     motifs = [phrase for _, phrase in matched_phrases]
-
+    darvo_score = calculate_darvo_score(
+        pattern_labels,
+        sentiment_before=0.0,
+        sentiment_after=sentiment_score,
+        motifs_found=motifs,
+        contradiction_flag=contradiction_flag
+)
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
         outputs = model(**inputs)
@@ -171,7 +232,7 @@ def analyze_single_message(text, thresholds):
     print(f"Contradiction: {contradiction_flag}")
     print("------------------\n")
 
-    return abuse_score, threshold_labels, top_patterns, result, stage
+    return abuse_score, threshold_labels, top_patterns, result, stage, darvo_score
 
 def analyze_composite(msg1, msg2, msg3, *answers_and_none):
     responses = answers_and_none[:len(ESCALATION_QUESTIONS)]