Update Gradio_UI.py

Gradio_UI.py

@@ -154,7 +154,11 @@ def stream_to_gradio(
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
+    print(f"DEBUG: Raw final_answer before handling: {final_answer}")
+    print(f"DEBUG: Type: {type(final_answer)}")
     final_answer = handle_agent_output_types(final_answer)
+    print(f"DEBUG: final_answer after handle_agent_output_types: {final_answer}")
+    print(f"DEBUG: Type after handling: {type(final_answer)}")
 
     if isinstance(final_answer, AgentText):
         yield gr.ChatMessage(
@@ -162,9 +166,15 @@ def stream_to_gradio(
             content=f"**Final answer:**\n{final_answer.to_string()}\n",
         )
     elif isinstance(final_answer, AgentImage):
+        print(f"DEBUG: Got AgentImage object: {final_answer}")
+        print(f"DEBUG: AgentImage type: {type(final_answer)}")
+        print(f"DEBUG: AgentImage attributes: {dir(final_answer)}")
+        image_path = final_answer.to_string()
+        print(f"DEBUG: Image path from to_string(): {image_path}")
+        print(f"DEBUG: Does path exist? {os.path.exists(image_path) if image_path else 'No path'}")
         yield gr.ChatMessage(
             role="assistant",
-            content={"path": final_answer.to_string(), "mime_type": "image/png"},
+            content={"path": image_path, "mime_type": "image/png"},
         )
     elif isinstance(final_answer, AgentAudio):
         yield gr.ChatMessage(