update

app.py

@@ -11,25 +11,37 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def generate_image(prompt: str, style: str) -> str:
+def generate_image(prompt: str, style: str = "") -> str:
     """A tool that generates an image based on a prompt and a style.
     Args:
         prompt: the prompt to generate the image
-        style: the style of the image
+        style: the style of the image (optional)
+    Returns:
+        An AgentImage object containing the generated image
     """
     try:
         # Add style to the prompt if specified
         full_prompt = f"{prompt}, {style} style" if style else prompt
 
-        # Generate the image
-        image = image_generation_tool(full_prompt)
+        # Generate the image using the loaded tool
+        # This should return a PIL image object
+        pil_image = image_generation_tool(full_prompt)
 
-        # Convert the PIL image to AgentImage
+        # Create a temporary file path to save the image
+        import os
+        import tempfile
+
+        # Create temp directory if it doesn't exist
+        temp_dir = os.path.join(tempfile.gettempdir(), "generated_images")
+        os.makedirs(temp_dir, exist_ok=True)
+
+        # Save the image to a file
+        img_path = os.path.join(temp_dir, f"generated_{hash(full_prompt)}.png")
+        pil_image.save(img_path)
+
+        # Return the AgentImage object with the file path
         from smolagents.agent_types import AgentImage
-        agent_image = AgentImage(image)
- 
-        # Return the AgentImage object
-        return agent_image
+        return AgentImage(img_path)
     except Exception as e:
         return f"Error generating image: {str(e)}"