Generate img

Gradio_UI.py

@@ -18,6 +18,7 @@ import os
 import re
 import shutil
 from typing import Optional
+from datetime import datetime
 
 from smolagents.agent_types import AgentAudio, AgentImage, AgentText, handle_agent_output_types
 from smolagents.agents import ActionStep, MultiStepAgent
@@ -122,6 +123,20 @@ def pull_messages_from_step(
         yield gr.ChatMessage(role="assistant", content=f"{step_footnote}")
         yield gr.ChatMessage(role="assistant", content="-----")
 
+def save_agent_image_to_tmp(agent_image):
+    """
+    Prend un AgentImage (smolagents), récupère l'image PIL
+    et la sauvegarde dans /tmp, puis renvoie le chemin.
+    """
+    pil_img = agent_image.to_pil()  # ou agent_image.image si l'attribut existe
+
+    ts = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
+    filename = f"agent_image_{ts}.webp"
+    filepath = os.path.join("/tmp", filename)
+
+    pil_img.save(filepath, format="WEBP")
+    return filepath
+
 
 def stream_to_gradio(
     agent,
@@ -170,13 +185,17 @@ def stream_to_gradio(
         # mais pour l’instant on laisse image_path = None.
 
     elif isinstance(final_answer, AgentImage):
-        # On suppose que to_string() renvoie un chemin de fichier image
-        image_path = final_answer.to_string()
+        # On convertit l'AgentImage en fichier sur disque
+        image_path = save_agent_image_to_tmp(final_answer)
+
         final_message = gr.ChatMessage(
             role="assistant",
             content=f"Image générée : {image_path}",
         )
 
+        yield final_message, image_path
+        return  # pour éviter de retomber sur le dernier yield en bas si tu l'as gardé
+
     elif isinstance(final_answer, AgentAudio):
         final_message = gr.ChatMessage(
             role="assistant",