Extract the generated images so they get displayed in UI

Gradio_UI.py

@@ -154,6 +154,11 @@ def stream_to_gradio(
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
+
+    # Extract the actual answer from FinalAnswerStep if needed
+    if hasattr(final_answer, "final_answer"):
+        final_answer = final_answer.final_answer
+
     final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):
@@ -293,4 +298,4 @@ class GradioUI:
         demo.launch(debug=True, share=True, **kwargs)
 
 
-__all__ = ["stream_to_gradio", "GradioUI"]
+__all__ = ["stream_to_gradio", "GradioUI"]