Update app.py

app.py

@@ -69,6 +69,13 @@ class BasicAgent:
         
         messages = [HumanMessage(content=question)]
         response = self.agent.invoke({"messages": messages})
+        
+        # Access final answer correctly
+        if isinstance(response, dict) and "messages" in response:
+            answer = response["messages"][-1].content
+        else:
+            answer = response.content
+            
         print(f"Agent returning fixed answer: {response.content}")
         return answer