Update app.py

app.py

@@ -22,12 +22,17 @@ class BasicAgent:
         print("BasicAgent initialized.")
         
     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 = main(question)
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        print(f"Agent received question: {question}")
+        
+        # Get the full response from your agent
+        full_response = main(question)
+        
+        # Extract just the answer part after "FINAL ANSWER: "
+        if "FINAL ANSWER: " in full_response:
+            return full_response.split("FINAL ANSWER: ", 1)[1].strip()
+        else:
+            return full_response.strip()
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):