Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import inspect
 import time
 import json
+import re
 import pandas as pd
 from langgraph.prebuilt import ToolNode, tools_condition
 from langgraph.graph.message import add_messages
@@ -68,7 +69,12 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer
+        m = re.search(r"FINAL ANSWER:\s*(.*)", answer, flags=re.DOTALL)
+        if m:
+            result = m.group(1)
+        else:
+            result = ""
+        return result
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):