Update app.py

app.py

@@ -122,40 +122,40 @@ 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
+# OLD VERSION
 def run_agent(question):
     try:
-        # Get all steps from the agent run
-        steps = list(agent.run(question, stream=False))
-
-        # Look through all tool calls to find final_answer
-        for step in steps:
-            if hasattr(step, "tool_calls") and step.tool_calls:
-                for call in step.tool_calls:
-                    if call.name == "final_answer":
-                        answer = call.arguments.get("answer", None)
-                        if answer:
-                            return [str(answer)]
-
-        # Fallback: try tool_output or .final_answer if FinalAnswerStep
-        for step in reversed(steps):
-            if hasattr(step, "tool_output") and step.tool_output:
-                return [str(step.tool_output)]
-            if hasattr(step, "final_answer") and step.final_answer:
-                return [str(step.final_answer)]
-
-        return ["null"]
+        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:
+#         # Get all steps from the agent run
+#         steps = list(agent.run(question, stream=False))
+
+#         # Look through all tool calls to find final_answer
+#         for step in steps:
+#             if hasattr(step, "tool_calls") and step.tool_calls:
+#                 for call in step.tool_calls:
+#                     if call.name == "final_answer":
+#                         answer = call.arguments.get("answer", None)
+#                         if answer:
+#                             return [str(answer)]
+
+#         # Fallback: try tool_output or .final_answer if FinalAnswerStep
+#         for step in reversed(steps):
+#             if hasattr(step, "tool_output") and step.tool_output:
+#                 return [str(step.tool_output)]
+#             if hasattr(step, "final_answer") and step.final_answer:
+#                 return [str(step.final_answer)]
+
+#         return ["null"]
+#     except Exception as e:
+#         return [f"Error: {e}"]
+
 
 
 demo = gr.Interface(fn=run_agent, inputs="text", outputs="text")