Update app.py

app.py

@@ -12,27 +12,23 @@ import requests
 from io import BytesIO
 from PIL import Image
 
-import base64
-
 @tool
-def generate_image(imageprompt: str) -> str:
+def generate_image(imageprompt: str) -> Image.Image:
     """A tool that generates an image based on a given prompt.
 
     Args:
         imageprompt: The description of the image to generate.
 
     Returns:
-        A base64 string representing the generated image.
+        The generated image as a PIL Image object.
     """
     url = f"https://image.pollinations.ai/prompt/{imageprompt}"
     response = requests.get(url)
     
     if response.status_code == 200:
-        image_data = BytesIO(response.content)
-        base64_image = base64.b64encode(image_data.getvalue()).decode("utf-8")
-        return f"data:image/png;base64,{base64_image}"
+        return Image.open(BytesIO(response.content))
     else:
-        return "Failed to generate image."
+        raise Exception("Failed to generate image.")
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: