Update app.py

app.py

@@ -122,12 +122,27 @@ 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}"]
+
+# NEW VERSION
 def run_agent(question):
     try:
-        result = agent(question)
-        return [str(result)]  # Must return a list with one string (like ["answer"])
+        steps = list(agent.run(question))
+        for step in reversed(steps):
+            if hasattr(step, "tool_calls"):
+                for call in step.tool_calls:
+                    if call.name == "final_answer":
+                        return [str(call.arguments.get("answer", "null"))]
+        return ["null"]  # fallback
     except Exception as e:
         return [f"Error: {e}"]
 
+
 demo = gr.Interface(fn=run_agent, inputs="text", outputs="text")
 demo.launch()