Handle final answer

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
+import re
 import requests
-import inspect
 import pandas as pd
 from langchain_core.messages import HumanMessage
 
@@ -28,11 +28,14 @@ class BasicAgent:
 
             last_message = result["messages"][-1].content
 
-            if "final answer" in last_message.lower():
-                split_final_answer = last_message.lower().split("final answer")
-                final_answer = split_final_answer[1].strip()
+            match = re.search(r"(?i)final\s*answer\s*[:\-]?\s*(.+)", last_message, re.DOTALL)
+            if match:
+                final_answer = match.group(1).strip()
             else:
-                final_answer = last_message
+                # Fallback: Use last non-empty line
+                lines = [line.strip() for line in last_message.split("\n") if line.strip()]
+                final_answer = lines[-1] if lines else last_message.strip()
+
             return final_answer
         except Exception as e: 
             print("Agent processing failed")