Update app.py

app.py

@@ -133,16 +133,30 @@ import gradio as gr
 # NEW VERSION
 def run_agent(question):
     try:
-        steps = list(agent.run(question))
-        for step in reversed(steps):
-            if hasattr(step, "tool_calls"):
+        # 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":
-                        return [str(call.arguments.get("answer", "null"))]
-        return ["null"]  # fallback
+                        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")
 demo.launch()