Update app.py

app.py

@@ -122,13 +122,13 @@ agent = CodeAgent(
 
 import gradio as gr
 
-# OLD VERSION
-def run_agent(question):
-    try:
-        result = agent(question)
-        return [str(result)]  # Must return a list with one string (like ["answer"])
-    except Exception as e:
-        return [f"Error: {e}"]
+# # OLD VERSION
+# def run_agent(question):
+#     try:
+#         result = agent(question)
+#         return [str(result)]  # Must return a list with one string (like ["answer"])
+#     except Exception as e:
+#         return [f"Error: {e}"]
 
 # # NEW VERSION
 # def run_agent(question):
@@ -156,6 +156,23 @@ def run_agent(question):
 #     except Exception as e:
 #         return [f"Error: {e}"]
 
+# NEWER VERSION (should just return the result)
+import re
+
+def run_agent(question):
+    try:
+        result = agent(question)
+
+        # If result is a string and contains "### 1. Task outcome", extract that
+        if isinstance(result, str):
+            match = re.search(r"### 1\. Task outcome \(short version\):\s*(.+)", result)
+            if match:
+                return [match.group(1).strip()]  # return just the short answer
+            return [result.strip()]
+
+        return [str(result)]
+    except Exception as e:
+        return [f"Error: {e}"]
 
 
 demo = gr.Interface(fn=run_agent, inputs="text", outputs="text")