Update Gradio_UI.py

Gradio_UI.py

@@ -153,7 +153,12 @@ def stream_to_gradio(
         ):
             yield message
 
-    final_answer = step_log  # Last log is the run's final_answer
+
+    if hasattr(step_log, "final_answer"):
+        final_answer = step_log.final_answer
+    else:
+        final_answer = step_log
+
     final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):