Update app.py

app.py

@@ -12,13 +12,16 @@ from Gradio_UI import GradioUI
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 from smolagents.agent_types import AgentImage  # 🔑 Импортируем AgentImage
 
+from smolagents.agent_types import AgentImage
+from PIL import Image
+
 @tool
 def generate_image(
     prompt: str, 
     negative_prompt: str = "blurry, text, watermark, low quality", 
     width: int = 1024, 
     height: int = 1024
-) -> AgentImage:  # 🔑 Указываем тип возвращаемого значения AgentImage
+) -> AgentImage:
     """
     Генерирует изображение по текстовому описанию.
     
@@ -38,7 +41,7 @@ def generate_image(
     from huggingface_hub import InferenceClient
     from PIL import Image
     import io
-    import os, tempfile, sys
+    import os, sys
     
     print(f"🔍 generate_image: prompt='{prompt[:50]}...', size={width}x{height}", file=sys.stderr)
     
@@ -55,21 +58,20 @@ def generate_image(
             guidance_scale=3.5
         )
         
-        # Получаем PIL объект
+        # 1. Получаем PIL объект
         if isinstance(result, Image.Image):
             image = result
         else:
             image = Image.open(io.BytesIO(result))
-        
-        # 🔑 Возвращаем сохранение во временный файл
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            image.save(tmp.name, format="PNG")
-            file_path = tmp.name
             
-        print(f"✅ generate_image: сохранено в {file_path}", file=sys.stderr)
+        # 2. 🔑 Форсируем загрузку пикселей в оперативную память!
+        # Это решает проблему "ленивой загрузки" PIL и предотвращает размер 0x0
+        image.load() 
+        
+        print(f"✅ generate_image: изображение {image.size} успешно сгенерировано!", file=sys.stderr)
         
-        # 🔑 Возвращаем AgentImage, передав ему путь к сохраненному файлу!
-        return AgentImage(file_path)
+        # 3. 🔑 Передаем в AgentImage САМ ОБЪЕКТ КАРТИНКИ (не путь к файлу!)
+        return AgentImage(image)
         
     except Exception as e:
         print(f"❌ generate_image exception: {e}", file=sys.stderr)
@@ -113,7 +115,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_image], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,