Update app.py

app.py

@@ -203,6 +203,17 @@ class BasicAgent:
             print(f"\nLLM Error: {str(e)}")
             raise
 
+    def _safe_parse(self, raw: str) -> dict:
+        """Fallback parser for when JSON parsing fails."""
+        try:
+            # Try to extract a dict-like structure
+            match = re.search(r'\{.*\}', raw, re.DOTALL)
+            if match:
+                return ast.literal_eval(match.group(0))
+        except:
+            pass
+        return {"needs_search": True, "search_query": ""}
+
     def _analyze_question(self, state: AgentState) -> AgentState:
         # Check for file attachments
         if state["file_url"]:
@@ -217,13 +228,18 @@ class BasicAgent:
 
         # Regular text question analysis
         prompt = (
-            "Decide if this question needs web search. Respond with a Python dict:\n"
-            "{\n  'needs_search': bool,\n  'search_query': str\n}\n\n"
+            "Return ONLY valid JSON:\n"
+            "{\"needs_search\": bool, \"search_query\": str}\n\n"
             f"Question: {state['question']}"
         )
         try:
             raw = self._call_llm(prompt)
-            decision = ast.literal_eval(raw)
+            try:
+                decision = json.loads(raw)
+            except json.JSONDecodeError:
+                print(f"JSON parse error, falling back to safe parse. Raw response: {raw}")
+                decision = self._safe_parse(raw)
+            
             state["needs_search"] = bool(decision.get("needs_search", False))
             state["search_query"] = decision.get("search_query", state["question"])
         except Exception as e: