Update app.py

app.py

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