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,23 +156,23 @@ import gradio as gr
 #     except Exception as e:
 #         return [f"Error: {e}"]
 
-# NEWER VERSION (should just return the result)
-import re
+# # NEWER VERSION (should just return the result)
+# import re
 
-def run_agent(question):
-    try:
-        result = agent(question)
+# 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()]
+#         # 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}"]
+#         return [str(result)]
+#     except Exception as e:
+#         return [f"Error: {e}"]
 
 
 demo = gr.Interface(fn=run_agent, inputs="text", outputs="text")