Update context_parser.py

context_parser.py

@@ -1,65 +1,82 @@
 from __future__ import annotations
 
 import re
-from typing import Dict, Optional
 
 
-def split_unity_message(full_text: str) -> tuple[str, str]:
-    if not full_text:
-        return "", ""
-    marker = "USER_MESSAGE:"
-    idx = full_text.find(marker)
-    if idx == -1:
-        return "", full_text.strip()
-    hidden = full_text[:idx].strip()
-    user = full_text[idx + len(marker):].strip()
-    return hidden, user
+def detect_intent(text: str) -> str:
+    t = (text or "").strip().lower()
 
+    if not t:
+        return "answer"
 
-def extract_game_context_fields(hidden_context: str) -> Dict[str, str]:
-    fields = {"category": "", "difficulty": "", "question": "", "options": ""}
-    if not hidden_context:
-        return fields
+    # definition
+    if (
+        re.search(r"\bdefine\b", t)
+        or re.search(r"\bdefinition\b", t)
+        or re.search(r"\bwhat does\b", t)
+        or re.search(r"\bwhat is meant by\b", t)
+    ):
+        return "definition"
 
-    patterns = {
-        "category": r"Category:\s*(.+)",
-        "difficulty": r"Difficulty:\s*(.+)",
-        "question": r"Question:\s*(.+?)(?:\nOptions:|\nPlayer balance:|\nLast outcome:|$)",
-        "options": r"Options:\s*(.+?)(?:\nPlayer balance:|\nLast outcome:|$)",
-    }
-    for key, pattern in patterns.items():
-        m = re.search(pattern, hidden_context, re.DOTALL)
-        if m:
-            fields[key] = m.group(1).strip()
-    return fields
+    # hint
+    if re.search(r"\bhint\b", t) or re.search(r"\bclue\b", t) or re.search(r"\bnudge\b", t):
+        return "hint"
 
+    # step/instruction
+    if (
+        re.search(r"\bfirst step\b", t)
+        or re.search(r"\bnext step\b", t)
+        or re.search(r"\bwhat should i do first\b", t)
+        or re.search(r"\bgive me the first step\b", t)
+    ):
+        return "instruction"
 
-def detect_intent(user_text: str, supplied: Optional[str] = None) -> str:
-    lower = (user_text or "").strip().lower()
+    # walkthrough / step-by-step
+    if (
+        re.search(r"\bwalk ?through\b", t)
+        or re.search(r"\bstep by step\b", t)
+        or re.search(r"\bfull working\b", t)
+        or re.search(r"\bwork through\b", t)
+    ):
+        return "walkthrough"
 
-    if supplied:
-        supplied = supplied.strip().lower()
-        if supplied in {"hint", "method", "walkthrough", "step_by_step", "full_working", "answer"}:
-            return supplied
+    # explain
+    if re.search(r"\bexplain\b", t) or re.search(r"\bwhy\b", t):
+        return "explain"
 
-    if any(p in lower for p in ["full working", "full working out", "show all the working", "complete working"]):
-        return "full_working"
-    if any(p in lower for p in ["step by step", "walkthrough", "work through", "explain step by step"]):
-        return "step_by_step"
-    if any(p in lower for p in ["how do i solve", "how to solve", "method", "what method", "approach"]):
+    # method
+    if (
+        re.search(r"\bmethod\b", t)
+        or re.search(r"\bapproach\b", t)
+        or re.search(r"\bhow do i solve\b", t)
+        or re.search(r"\bhow to solve\b", t)
+    ):
         return "method"
-    if any(p in lower for p in ["hint", "nudge", "first step", "how do i start", "what do i do first"]):
-        return "hint"
-    if any(p in lower for p in ["why", "explain", "breakdown"]):
-        return "step_by_step"
-    if any(p in lower for p in ["solve", "what is", "answer", "give me the answer"]):
+
+    # concept
+    if re.search(r"\bconcept\b", t) or re.search(r"\bprinciple\b", t) or re.search(r"\brule\b", t):
+        return "concept"
+
+    # plain answer / direct solve
+    if (
+        re.search(r"\bsolve\b", t)
+        or re.search(r"\bwhat is\b", t)
+        or re.search(r"\bfind\b", t)
+        or re.search(r"\bgive (?:me )?the answer\b", t)
+        or re.search(r"\bjust the answer\b", t)
+        or re.search(r"\banswer only\b", t)
+        or re.search(r"\bcalculate\b", t)
+    ):
         return "answer"
+
     return "answer"
 
 
 def intent_to_help_mode(intent: str) -> str:
+    if intent in {"walkthrough", "step_by_step", "explain", "method", "concept"}:
+        return "walkthrough"
     if intent == "hint":
         return "hint"
-    if intent in {"method", "step_by_step", "full_working"}:
-        return "walkthrough"
-    return "answer"
+    if intent in {"definition", "instruction"}:
+        return intent
+    return "answer"