Update Gradio_UI.py

Gradio_UI.py

@@ -155,6 +155,12 @@ def stream_to_gradio(
 
     final_step = step_log  # this is the final ActionStep
     final_output = getattr(final_step, "tool_output", None)
+
+    # ✅ 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)
     
     if isinstance(final_output, AgentText):