Show images as images

Gradio_UI.py

@@ -162,9 +162,14 @@ def stream_to_gradio(
             content=f"**Final answer:**\n{final_answer.to_string()}\n",
         )
     elif isinstance(final_answer, AgentImage):
+        # yield gr.ChatMessage(
+        #     role="assistant",
+        #     content={"path": final_answer.to_string(), "mime_type": "image/png"},
+        # )
+        image_path = final_answer.to_string()
         yield gr.ChatMessage(
-            role="assistant",
-            content={"path": final_answer.to_string(), "mime_type": "image/png"},
+        role="assistant",
+        content=f"**Final answer:**\n![Generated Image]({image_path})",
         )
     elif isinstance(final_answer, AgentAudio):
         yield gr.ChatMessage(