Update app.py

app.py

@@ -270,16 +270,29 @@ class BasicAgent:
 
 
     def __call__(self, question: str) -> str:
+        print(f"Agent received question: {question[:50]}{'...' if len(question) > 50 else ''}")
+        
+        # Initialize state with the question
         state = init_state(question)
+        
+        # Execute the workflow
         final_state = self.workflow.invoke(state)
         
-        # Extract final answer
+        # Debug: Print the final state structure
+        print(f"Final state keys: {list(final_state.keys())}")
+        if 'history' in final_state:
+            print(f"History length: {len(final_state['history'])}")
+            for i, msg in enumerate(final_state['history']):
+                print(f"Message {i}: {type(msg).__name__} - {msg.content[:100]}...")
+        
+        # Extract final answer from history
         for msg in reversed(final_state['history']):
             if isinstance(msg, AIMessage) and "FINAL ANSWER:" in msg.content:
-                return msg.content.split("FINAL ANSWER:")[1].strip()
-        
-        return "No final answer found"
-
+                # Extract and clean the final answer
+                answer = msg.content.split("FINAL ANSWER:")[1].strip()
+                print(f"Agent returning answer: {answer}")
+                return answer
+    
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):