update app

app.py

@@ -24,24 +24,14 @@ def generate_image(prompt: str, style: str = "") -> str:
         full_prompt = f"{prompt}, {style} style" if style else prompt
 
         # Generate the image using the loaded tool
-        # This should return a PIL image object
         pil_image = image_generation_tool(full_prompt)
 
-        # 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
+        # Convert the PIL image to AgentImage
         from smolagents.agent_types import AgentImage
-        return AgentImage(img_path)
+        agent_image = AgentImage(pil_image)
+        
+        # Return the AgentImage object
+        return agent_image
     except Exception as e:
         return f"Error generating image: {str(e)}"