update

pydvpl_bot.py

@@ -19,6 +19,7 @@ from craiyon import Craiyon, craiyon_utils
 from io import BytesIO
 import base64
 import craiyon
+import aiohttp
 
 
 
@@ -589,21 +590,24 @@ async def image(ctx, *, prompt: str):
     await ctx.send(f"Generating images for prompt: `{prompt}`. Please be patient...")
 
     try:
-        generator = Craiyon()
-        generated_images = await generator.async_generate(prompt)
+        headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36'}
 
-        # Encode images to base64
-        b64_list = await craiyon_utils.async_encode_base64(generated_images.images)
+        async with aiohttp.ClientSession(headers=headers) as session:
+            generator = Craiyon(session=session)
+            generated_images = await generator.async_generate(prompt)
 
-        # Prepare images for sending
-        images = []
-        for index, image_b64 in enumerate(b64_list):
-            img_bytes = BytesIO(base64.b64decode(image_b64))
-            image_file = discord.File(img_bytes, filename=f"result{index}.webp")
-            images.append(image_file)
+            # Encode images to base64
+            b64_list = await craiyon_utils.async_encode_base64(generated_images.images)
 
-        # Send images to the user
-        await ctx.reply(files=images)
+            # Prepare images for sending
+            images = []
+            for index, image_b64 in enumerate(b64_list):
+                img_bytes = BytesIO(base64.b64decode(image_b64))
+                image_file = discord.File(img_bytes, filename=f"result{index}.webp")
+                images.append(image_file)
+
+            # Send images to the user
+            await ctx.reply(files=images)
 
     except Exception as e:
         # Print the specific exception message for debugging