Tool img

app.py

@@ -8,7 +8,7 @@ from tools.final_answer import FinalAnswerTool
 #from tools.web_search import DuckDuckGoSearchTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.http_search import HttpSearchTool
-from tools.image_wrapper import generate_image_and_save
+from tools.image_wrapper import save_image_to_tmp
 
 from Gradio_UI import GradioUI
 
@@ -39,6 +39,17 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Erreur pour le fuseau '{timezone}' : {str(e)}"
 
+@tool
+def generate_image_and_save(prompt: str) -> str:
+    """Génère une image via image_generation_tool, la sauvegarde dans /tmp
+    et renvoie le chemin du fichier.
+    Args:
+        prompt: description textuelle de l'image à générer.
+    """
+    img = image_generation_tool(prompt=prompt)
+    filepath = save_image_to_tmp(img)
+    return f"Image sauvegardée dans : {filepath}"
+
 
 # ==== Instanciation des tools existants ====