Img display tool

app.py

@@ -50,6 +50,16 @@ def generate_image_and_save(prompt: str) -> str:
     filepath = save_image_to_tmp(img)
     return f"Image sauvegardée dans : {filepath}"
 
+@tool
+def generate_image(prompt: str):
+    """Génère une image à partir d'un prompt et la renvoie directement.
+    L'UI Gradio pourra l'afficher.
+    Args:
+        prompt: description textuelle de l'image.
+    """
+    img = image_generation_tool(prompt=prompt)
+    return img
+
 
 # ==== Instanciation des tools existants ====
 
@@ -88,6 +98,7 @@ agent = CodeAgent(
         visit_webpage,              # lecture de page web
         image_generation_tool,      # génération d’image
         generate_image_and_save,
+        generate_image,
         my_custom_tool,             # tool custom
         get_current_time_in_timezone,
         final_answer,               # final_answer doit être présent