Update app.py

app.py

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