Update motif_tagging.py

motif_tagging.py

@@ -9,7 +9,7 @@ MOTIF_PATTERNS = {
         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 rose garden\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",
@@ -34,9 +34,10 @@ def detect_motifs(text):
 
     for label, patterns in MOTIF_PATTERNS.items():
         for pattern in patterns:
-            if re.search(pattern, text_lower):
+            match = re.search(pattern, text_lower)
+            if match:
                 tags.append(label)
-                matched_phrases.append((label, pattern))
+                matched_phrases.append((label, match.group()))
                 break
 
     return tags, matched_phrases