Update motif_tagging.py

motif_tagging.py

@@ -3,27 +3,55 @@
 import re
 
 MOTIF_PATTERNS = {
-    "physical_threat": [
-        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",
-        r"\b(i'll|i will|ill) put you in (a|the) rose garden\b"
+"physical_threat": [
+    r"\b(i['’]m|i am) going to (hurt|kill|break|ruin|destroy|end|bash|beat|smash|wreck) you\b",
+    r"\b(i['’]?ll|i will) (hurt|kill|ruin|wreck|end|destroy|obliterate|crush|eliminate|beat) you\b",
+    r"\b(i['’]?ll|i will) make you disappear\b",
+    r"\b(i['’]?ll|i will) put you in (a|the) (grave|hospital|ground|dirt|bodybag)\b",
+    r"\b(i['’]?ll|i will) snap your (neck|spine|arm|leg)\b",
+    r"\b(i['’]?ll|i will) break your (face|jaw|legs|arms|neck)\b",
+    r"\b(i['’]?ll|i will) beat the (shit|hell|crap) out of you\b",
+    r"\b(i['’]?ll|i will) f\*?ck you up\b",
+    r"\b(i['’]?ll|i will) (come for|find|hunt|get) you\b",
+    r"\b(i['’]?ll|i will) put you down\b",
+    r"\b(i['’]?ll|i will) end you\b",
+    r"\bi hope you (die|suffer|disappear)\b",
+    r"\byou better (run|hide|watch your back)\b",
+    r"\bsay goodbye\b.*\byou\b",
+    r"\bsleep with one eye open\b",
+    r"\bi['’]?ll|i will) send you to (hell|the morgue)\b",
+    r"\bi['’]?ll|i will) make sure you’re never seen again\b"
+    r"\b(i'll|i will|ill) put you in (a|the) rose garden\b"
 ],
     "extreme_control": [
-        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",
-],
+    r"\bi (won[’']?t|will not) let you (leave|go|talk to|see|text|spend time with) (him|her|them|anyone)\b",
+    r"\byou (can[’']?t|are not allowed to|shouldn’t) (go out|leave|talk to|see|text|spend time with|have friends)\b",
+    r"\bi (decide|get to decide|control|own) what you (do|say|wear|eat|spend|see|talk to)\b",
+    r"\b(you['’]?re|you are) not allowed to (leave|talk to anyone|have your phone)\b",
+    r"\bi make the rules\b",
+    r"\bit’s my way or nothing\b",
+    r"\bi’ll take your (car|keys|money|phone)\b",
+    r"\byou don’t need anyone else\b",
+    r"\byou belong to me\b",
+    r"\bwithout me, you’re (nothing|no one)\b",
+    r"\byou do what i say\b",
+    r"\b(you better|you have to) do what i say\b",
+    r"\b(you['’]?re not going|you’re not leaving|i won’t let you go)\b"
+]
 
-    "suicidal_threat": [
-        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",
-],
+
+"suicidal_threat": [
+    r"\b(i['’]?ll|i will) (kill|hurt) myself (if you leave|if you don’t come back|unless you stay)\b",
+    r"\bi (don’t|do not) want to live (without you|if you leave)\b",
+    r"\bi['’]?ll (die|disappear|end it) (if|unless) you (leave|go|don’t call)\b",
+    r"\b(i['’]?ll|i will) make you watch me (die|bleed|suffer)\b",
+    r"\bi['’]?ll do something bad (to myself|if you leave)\b",
+    r"\bi['’]?ll regret it when i’m gone\b",
+    r"\bi can’t live without you\b",
+    r"\bmaybe (i should just die|it’d be better if i were dead)\b",
+    r"\bi’m better off dead\b",
+    r"\bi['’]?ll disappear forever if you walk out\b"
+]
 }