Update app.py

app.py

@@ -268,13 +268,17 @@ class BasicAgent:
         ]
     
 
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # Wrap the question in a HumanMessage from langchain_core
-        messages = [HumanMessage(content=question)]
-        messages = self.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        state = init_state(question)
+        final_state = self.workflow.invoke(state)
+        
+        # Extract final answer
+        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"