Update app.py

app.py

@@ -17,15 +17,19 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 # Image generation !
 
+@tool
 def generate_image_tool(prompt: str) -> str:
     """Generates an image from text and returns the file path."""
     try:
-        image = image_generation_tool(prompt) 
-        return image  # Return the file path
+        image = image_generation_tool(prompt)  # Get PIL Image
+        file_path = "generated_image.jpg"
+        image.save(file_path)  # Save it
+        return file_path  # Return the file path
     except Exception as e:
         return f"Error generating image: {str(e)}"
 
 
+
 # @tool
 # def generate_image_tool(prompt: str) -> Image:
 #     """Generates an image from text using the image generation tool.