Update code

scope_router.py

@@ -72,6 +72,48 @@ CODE_PATTERNS = [
     r"\bsyntax\b",
     r"\bprogram\b",
     r"\bscript\b",
+    r"\bsoftware\b",
+    r"\bbackend\b",
+    r"\bfrontend\b",
+    r"\bendpoint\b",
+    r"\bdatabase\b",
+    r"\bquery\b",
+    r"\btraceback\b",
+    r"\bdocstring\b",
+    r"\btype hints?\b",
+    r"\bvalidation\b",
+    r"\berror handling\b",
+    r"\blogging\b",
+    r"\bcomments\b",
+    r"\bexception handling\b",
+]
+
+CODE_MODIFICATION_PATTERNS = [
+    r"\badd\b",
+    r"\bupdate\b",
+    r"\bmodify\b",
+    r"\bchange\b",
+    r"\brename\b",
+    r"\bconvert\b",
+    r"\bextend\b",
+    r"\binclude\b",
+    r"\buse\b",
+    r"\breplace\b",
+    r"\bremove\b",
+    r"\bdelete\b",
+    r"\bdocstring\b",
+    r"\btype hints?\b",
+    r"\bvalidation\b",
+    r"\berror handling\b",
+    r"\blogging\b",
+    r"\bcomments\b",
+    r"\bexception handling\b",
+    r"\bmake it\b",
+    r"\bturn this into\b",
+    r"\badd support for\b",
+    r"\bnow add\b",
+    r"\bnow update\b",
+    r"\bnow change\b",
 ]
 
 IMAGE_PATTERNS = [
@@ -158,6 +200,7 @@ def looks_like_code_block(text: str) -> bool:
         "=>",
         "public ",
         "private ",
+        "protected ",
         "const ",
         "let ",
         "var ",
@@ -166,12 +209,43 @@ def looks_like_code_block(text: str) -> bool:
         "for (",
         "while (",
         "</",
+        "try:",
+        "except:",
     ]
 
     lowered = text.lower()
     return any(signal.lower() in lowered for signal in code_signals)
 
 
+def is_follow_up_code_request(request: CodeXRequest) -> bool:
+    message = normalize_text(request.message)
+
+    has_session = bool(request.session_id and str(request.session_id).strip())
+    has_previous_context = bool(request.previous_context and str(request.previous_context).strip())
+
+    if not (has_session or has_previous_context):
+        return False
+
+    if contains_pattern(message, CODE_MODIFICATION_PATTERNS):
+        return True
+
+    short_follow_up_phrases = [
+        "now add",
+        "now update",
+        "now change",
+        "now remove",
+        "now use",
+        "add this",
+        "update this",
+        "change this",
+        "fix this",
+        "refactor this",
+        "review this",
+    ]
+
+    return any(phrase in message for phrase in short_follow_up_phrases)
+
+
 def is_code_related_request(request: CodeXRequest) -> bool:
     message = normalize_text(request.message)
     code = request.code or ""
@@ -192,6 +266,9 @@ def is_code_related_request(request: CodeXRequest) -> bool:
     if contains_pattern(message, CODE_PATTERNS):
         return True
 
+    if is_follow_up_code_request(request):
+        return True
+
     return False