Img view

tools/image_wrapper.py

@@ -1,27 +1,32 @@
 from smolagents import tool
 import os
 from datetime import datetime
-from app import image_generation_tool
 
-# image_generation_tool est déjà chargé via load_tool(...)
+# Ce wrapper NE connaît PAS image_generation_tool.
+# On lui passera la fonction au moment de l'appel.
 
-@tool
-def generate_image_and_save(prompt: str) -> str:
-    """Génère une image à partir d'un prompt, la sauvegarde dans /tmp
-    et renvoie le chemin du fichier enregistré.
-    Args:
-        prompt: description textuelle de l'image à générer.
-    """
-    # 1. Génération de l'image via le tool HF
-    img = image_generation_tool(prompt=prompt)
 
-    # 2. Création d'un nom de fichier unique
+def _save_image_to_tmp(img) -> str:
+    """Sauvegarde une image PIL dans /tmp et renvoie le chemin."""
     ts = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
     filename = f"generated_image_{ts}.webp"
     filepath = os.path.join("/tmp", filename)
-
-    # 3. Sauvegarde (img est un objet PIL.Image.Image)
     img.save(filepath, format="WEBP")
+    return filepath
+
+
+@tool
+def generate_image_and_save(prompt: str, _image_tool=None) -> str:
+    """Génère une image via _image_tool, la sauvegarde dans /tmp et renvoie le chemin.
+    Args:
+        prompt: Description textuelle de l'image à générer.
+        _image_tool: fonction/tool à appeler, injectée depuis app.py.
+    """
+    if _image_tool is None:
+        return (
+            "Erreur: aucun tool d'image n'a été fourni à generate_image_and_save."
+        )
 
-    # 4. Retour d'un chemin affichable par Gradio
+    img = _image_tool(prompt=prompt)
+    filepath = _save_image_to_tmp(img)
     return f"Image sauvegardée dans : {filepath}"