update

pydvpl_bot.py

@@ -579,6 +579,9 @@ async def chat(ctx, model, *, message):
 # FUNCTION ------------------------------------------------------ SPLIT
 
 
+# Create a global aiohttp.ClientSession object
+session = aiohttp.ClientSession()
+
 @bot.hybrid_command(help="Generate images with 'craiyon' ai.")
 async def image(ctx, *, prompt: str):
     # Check if the user is authorized to use the command
@@ -590,24 +593,21 @@ async def image(ctx, *, prompt: str):
     await ctx.send(f"Generating images for prompt: `{prompt}`. Please be patient...")
 
     try:
-        headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36'}
-
-        async with aiohttp.ClientSession(headers=headers) as session:
-            generator = Craiyon(session=session)
-            generated_images = await generator.async_generate(prompt)
+        generator = Craiyon()  # Instantiate Craiyon without passing a session
+        generated_images = await generator.async_generate(prompt, session=session)  # Pass the session object here
 
-            # Encode images to base64
-            b64_list = await craiyon_utils.async_encode_base64(generated_images.images)
+        # Encode images to base64
+        b64_list = await craiyon_utils.async_encode_base64(generated_images.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)
+        # 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)
+        # Send images to the user
+        await ctx.reply(files=images)
 
     except Exception as e:
         # Print the specific exception message for debugging