Update app.py

app.py

@@ -82,9 +82,15 @@ class BasicAgent:
         self.tool_chain = build_tool_graph(self.agent_prompt)
 
     def format_final_answer(self, answer: str) -> str:
-        cleaned = " ".join(answer.split())
-        match = re.search(r"FINAL ANSWER:\s*(.*)", cleaned, re.IGNORECASE)
-        return f"FINAL ANSWER: {match.group(1).strip()}" if match else f"FINAL ANSWER: {cleaned}"
+        # Clean up whitespace
+        cleaned = " ".join(answer.strip().split())
+        # Extract only the final answer after the last occurrence of 'FINAL ANSWER:'
+        if "FINAL ANSWER:" in cleaned.upper():
+            final = re.split(r"FINAL ANSWER:\s*", cleaned, flags=re.IGNORECASE)[-1]
+        else:
+            final = cleaned
+        return f"FINAL ANSWER: {final.strip()}"
+
 
     def query_groq(self, question: str) -> str:
         full_prompt = f"{self.agent_prompt}\n\nQuestion: {question}"