Update app.py

app.py

@@ -116,7 +116,12 @@ Do not include any other text or explanation. Only return the dictionary.
 """
         
         try:
-            analysis = ast.literal_eval(self._call_llm_api(prompt))
+            llm_response = self._call_llm_api(prompt)
+            print("\n=== Analyze Question LLM Response ===")
+            print(f"Input: {state['question']}")
+            print(f"LLM Response: {llm_response}")
+            
+            analysis = ast.literal_eval(llm_response)
             state["needs_more_info"] = analysis.get('needs_search', False)
             state["search_query"] = analysis.get('search_query', '')
             
@@ -136,12 +141,19 @@ Do not include any other text or explanation. Only return the dictionary.
     def _use_search(self, state: AgentState) -> AgentState:
         """Use the search tool."""
         try:
+            print("\n=== Search Tool ===")
+            print(f"Search Query: {state['search_query']}")
+            
             # Use the simplified search function
             search_results = simple_search(
                 query=state["search_query"],
                 max_results=3
             )
             
+            print("Search Results:")
+            for i, result in enumerate(search_results, 1):
+                print(f"{i}. {result}")
+            
             state["history"].append({
                 'step': 'search',
                 'query': state["search_query"],
@@ -150,6 +162,7 @@ Do not include any other text or explanation. Only return the dictionary.
             state["needs_more_info"] = False
             state["current_step"] = 'final_answer'
         except Exception as e:
+            print(f"Search Error: {e}")
             state["history"].append({
                 'step': 'search_error',
                 'error': str(e)
@@ -170,7 +183,16 @@ Do not include any other text or explanation. Only return the dictionary.
         Provide a clear, concise answer that addresses the original question.
         """
         
-        state["final_answer"] = self._call_llm_api(prompt)
+        print("\n=== Generate Final Answer ===")
+        print(f"Question: {state['question']}")
+        print("History:")
+        print(history_str)
+        
+        llm_response = self._call_llm_api(prompt)
+        print("\nFinal Answer:")
+        print(llm_response)
+        
+        state["final_answer"] = llm_response
         return state
 
     def __call__(self, question: str) -> str: