Update app.py

app.py

@@ -149,7 +149,11 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = alfred.invoke({"messages": [system_prompt, question]},{"recursion_limit": 3})
+        answer = alfred.invoke({"messages": [system_prompt, question]},{"recursion_limit": 3})['messages'][-1].content
+        answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))    
+        answer = answer.replace('FINAL ANSWER: ', '')
+        answer = answer.replace('[', '')
+        fixed_answer = answer.replace(']', '')
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer