Fix image rendering in GradioUI and avoid forced share=True on Hugging Face Spaces

Gradio_UI.py

@@ -154,6 +154,9 @@ def stream_to_gradio(
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
+    if hasattr(final_answer, "final_answer"):
+        final_answer = final_answer.final_answer  # unwrap FinalAnswerStep
+
     final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):
@@ -162,9 +165,16 @@ def stream_to_gradio(
             content=f"**Final answer:**\n{final_answer.to_string()}\n",
         )
     elif isinstance(final_answer, AgentImage):
+        path = final_answer.to_string()
+        # If to_string() isn't a real file path, try saving the PIL image
+        if not isinstance(path, str) or not os.path.exists(path):
+            if hasattr(final_answer, "image") and final_answer.image is not None:
+                os.makedirs("outputs", exist_ok=True)
+                path = os.path.join("outputs", "generated.png")
+                final_answer.image.save(path)
         yield gr.ChatMessage(
             role="assistant",
-            content={"path": final_answer.to_string(), "mime_type": "image/png"},
+            content={"path": path, "mime_type": "image/png"},
         )
     elif isinstance(final_answer, AgentAudio):
         yield gr.ChatMessage(
@@ -290,7 +300,7 @@ class GradioUI:
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
 
-        demo.launch(debug=True, share=True, **kwargs)
+        demo.launch(debug=True, **kwargs)
 
 
 __all__ = ["stream_to_gradio", "GradioUI"]