Update app.py

app.py

@@ -35,11 +35,34 @@ MAX_TOOL_CALLS = 5
 # ─────────────────────────── Helper utilities ────────────────────────────
 
 def safe_json(text: str) -> Optional[Dict[str, Any]]:
-    try:
-        obj = json.loads(text.strip())
-        return obj if isinstance(obj, dict) else None
-    except json.JSONDecodeError:
-        return None
+    """Best‑effort extractor that tolerates code‑fences and leading commentary.
+    It returns the *first* top‑level JSON object found in the string or None.
+    """
+    import re, json
+
+    # Drop markdown code‑fences if present
+    if text.strip().startswith("```"):
+        parts = re.split(r"```+", text.strip(), maxsplit=2)
+        text = parts[1] if len(parts) > 1 else parts[0]
+
+    # Locate the first {...} block at top level
+    brace_stack, start = [], None
+    for i, ch in enumerate(text):
+        if ch == '{':
+            if not brace_stack:
+                start = i
+            brace_stack.append(ch)
+        elif ch == '}' and brace_stack:
+            brace_stack.pop()
+            if not brace_stack and start is not None:
+                candidate = text[start:i+1]
+                try:
+                    obj = json.loads(candidate)
+                    return obj if isinstance(obj, dict) else None
+                except json.JSONDecodeError:
+                    # Keep searching in case there is another block later
+                    start = None
+    return None
 
 
 # def brief(d: Dict[str, Any]) -> str:
@@ -75,9 +98,9 @@ def tool_selector(state: AgentState) -> AgentState:
     raw = LLM(state.messages + [prompt]).content.strip()
     print(f"Tool selector response: {raw}")
     state.add(AIMessage(content=raw))
-    # parsed = safe_json(raw)
-    parsed = json.loads(raw)
-    print("parsed : ", parsed)
+    parsed = safe_json(raw)
+    # parsed = json.loads(raw)
+    # print("parsed : ", parsed)
     print(f"Parsed: {parsed}")
     if not parsed or "action" not in parsed:
         state.next_action = "final"