Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+import ast
 from typing import Dict, Any, List, TypedDict, Optional
 from langgraph.graph import Graph, StateGraph
 from langgraph.prebuilt import ToolNode
@@ -99,21 +100,31 @@ class BasicAgent:
     def _analyze_question(self, state: AgentState) -> AgentState:
         """Analyze the question and determine the next step."""
         prompt = f"""Analyze this question and determine what needs to be done: {state['question']}
-        Return your analysis in this format:
-        {{
-            "needs_search": true/false,
-            "search_query": "query if needed"
-        }}
-        """
-        
-        analysis = eval(self._call_llm_api(prompt))
-        state["needs_more_info"] = analysis.get('needs_search', False)
-        state["search_query"] = analysis.get('search_query', '')
+
+Return only a valid Python dictionary in this exact format:
+{{
+    "needs_search": true/false,
+    "search_query": "query if needed"
+}}
+
+Do not include any other text or explanation. Only return the dictionary.
+"""
         
-        if analysis.get('needs_search', False):
+        try:
+            analysis = ast.literal_eval(self._call_llm_api(prompt))
+            state["needs_more_info"] = analysis.get('needs_search', False)
+            state["search_query"] = analysis.get('search_query', '')
+            
+            if analysis.get('needs_search', False):
+                state["current_step"] = 'search'
+            else:
+                state["current_step"] = 'final_answer'
+        except (ValueError, SyntaxError) as e:
+            print(f"Error parsing LLM response: {e}")
+            # Default to search if we can't parse the response
+            state["needs_more_info"] = True
+            state["search_query"] = state["question"]
             state["current_step"] = 'search'
-        else:
-            state["current_step"] = 'final_answer'
         
         return state