Update Gradio_UI.py

Gradio_UI.py

@@ -153,16 +153,17 @@ def stream_to_gradio(
         ):
             yield message
 
-    final_step = step_log  # this is the final ActionStep
-    final_output = getattr(final_step, "tool_output", None)
+    final_step = step_log  # last step returned by agent.run()
+    final_output = None
     
-    # If no tool_output, fallback to first final_answer call
-    if not final_output and final_step.tool_calls:
+    if hasattr(final_step, "tool_calls"):
         for call in final_step.tool_calls:
             if call.name == "final_answer":
-                final_output = call.arguments.get("answer")
+                # Prefer the actual output if available, fallback to arguments
+                final_output = call.output if call.output is not None else call.arguments.get("answer")
                 break
 
+
     
     final_output = handle_agent_output_types(final_output)