Update Gradio_UI.py

Gradio_UI.py

@@ -155,11 +155,14 @@ def stream_to_gradio(
 
     final_step = step_log  # this is the final ActionStep
     final_output = getattr(final_step, "tool_output", None)
+    
+    # If no tool_output, fallback to first final_answer call
+    if not final_output and final_step.tool_calls:
+        for call in final_step.tool_calls:
+            if call.name == "final_answer":
+                final_output = call.arguments.get("answer")
+                break
 
-    # ✅ Fallback if tool_output is None and a final_answer tool was called
-    if final_output is None and hasattr(final_step, "tool_calls") and final_step.tool_calls:
-        if final_step.tool_calls[0].name == "final_answer":
-            final_output = final_step.tool_calls[0].arguments.get("answer", None)
     
     final_output = handle_agent_output_types(final_output)