Update app.py

app.py

@@ -27,10 +27,8 @@ class BasicAgent:
         response = state_dict["messages"][-1]
         try:
             response = response.content
-            if "Final answer:" in response:
-                response = response.split("Final answer:")[1]
-            response = response.strip()
-            return response
+            matches = re.findall(r"Final Answer:\s*(.*)", response)
+            return matches[-1].strip() if matches else response.strip()
         except:
             return "Sorry, I can't answer to this question."