Update motif_tagging.py

motif_tagging.py

@@ -4,24 +4,25 @@ import re
 
 MOTIF_PATTERNS = {
     "physical_threat": [
-        r"\b(i'?ll|i am going to) (hurt|kill|break|end|ruin|destroy) you\b",
-        r"\bsay goodbye to (your|those)? (kneecaps|teeth|face)\b",
-        r"\bi'?ll put you in a (grave|hole|rose garden)\b",
-        r"\b(sleep with one eye open|you’ll see what happens)\b",
-        r"\bi'?ll make you disappear\b",
-    ],
+        r"\b(i am going to|i'll) (hurt|kill|break|end|ruin|destroy) you\b",
+        r"\bsay goodbye to (you|those)? (kneecaps|teeth|face)\b",
+        r"\b(i'll|i will) put you in a (grave|hole|rose garden)\b",
+        r"\bsleep with one eye open( you'll see what happens)?\b",
+        r"\b(i'll|i will) make you disappear\b",
+],
     "extreme_control": [
-        r"\bi decide who you (see|talk to|text|spend time with)\b",
-        r"\byou’re not allowed to\b",
-        r"\byou don’t get to (leave|say no|argue)\b",
+        r"\b(i decide|i control) who you (see|talk to|text|spend time with)\b",
+        r"\b(you('re| are) not allowed to)\b",
+        r"\byou (don't|do not) get to (leave|say no|argue)\b",
         r"\bi own you\b",
-    ],
+],
+
     "suicidal_threat": [
-        r"\bi'?ll kill myself\b",
-        r"\bi don’t want to live if you leave\b",
-        r"\bi’ll die without you\b",
-        r"\byou’ll regret it when i’m gone\b",
-    ],
+        r"\b(i'll|i will) kill myself\b",
+        r"\bi (don’t|do not) want to live if you leave\b",
+        r"\b(i’ll|i will) die without you\b",
+        r"\byou(’ll|'ll) regret it when i(’m|'m) gone\b",
+],
 }
 
 def tag_motifs(text):