update test

pydvpl_bot.py

@@ -15,9 +15,11 @@ import pytgpt.opengpt as opengpt
 import pytgpt.blackboxai as blackboxai
 import subprocess
 import pytgpt
+from craiyon import Craiyon, craiyon_utils
 from io import BytesIO
 import base64
-from craiyon import Craiyon, craiyon_utils
+import craiyon
+
 
 
 # ENV_LOADER ------------------------------------------------------ START
@@ -504,7 +506,7 @@ async def version(ctx):
     # Create an embed message to display the bot version
     embed = discord.Embed(
         title="Version Info",
-        description=f"The current version of libraries used.",
+        description=f"The current version of libraries used in `PyDVPL`.",
         color=discord.Color.yellow()
     )
 
@@ -513,6 +515,7 @@ async def version(ctx):
     embed.add_field(name="LZ4 PY Lib Version:", value=f"``{__LZ4_VERSION__}``", inline=False)
     embed.add_field(name="Discord PY Version:", value=f"``{discord.__version__}``", inline=False)
     embed.add_field(name="PyTGPT Lib Version:", value=f"``{pytgpt.__version__}``", inline=False)
+    embed.add_field(name="Craiyon Lib Version:", value=f"``{craiyon.__version__}``", inline=False)
     
     # Send the embed message
     await ctx.send(embed=embed) 
@@ -561,29 +564,33 @@ async def chat(ctx, model, *, message):
 
 @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
     if str(ctx.author.id) not in AUTHORIZED_AI_ACCESS_ID:
-        await ctx.send("You are not authorized to use this command. Only `@rifsxd`'s server boosters & special permitted users can enjoy free `ChatGPT` access.")
+        await ctx.send("You are not authorized to use this command.")
         return
 
-    await ctx.send(f"Generating prompt `{prompt}` please be patient...")
-    
+    # Inform the user that the generation process has started
+    await ctx.send(f"Generating images for prompt: `{prompt}`. Please be patient...")
+
     try:
         generator = Craiyon()
-        generated_images = await generator.async_generate(prompt)  # Generate images
-        b64_list = await craiyon_utils.async_encode_base64(generated_images.images)  # Download images from https://img.craiyon.com and store them as b64 bytestring object
+        generated_images = await generator.async_generate(prompt)
+
+        # Encode images to base64
+        b64_list = await craiyon_utils.async_encode_base64(generated_images.images)
 
-        images1 = []
-        for index, image in enumerate(b64_list):  # Loop through b64_list, keeping track of the index
-            img_bytes = BytesIO(base64.b64decode(image))  # Decode the image and store it as a bytes object
-            image = discord.File(img_bytes)
-            image.filename = f"result{index}.webp"
-            images1.append(image)  # Add the image to the images1 list
+        # 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)
 
-        await ctx.reply(files=images1)  # Reply to the user with all 9 images in 1 message
+        # Send images to the user
+        await ctx.reply(files=images)
 
     except Exception as e:
-        await ctx.send(f"An error occurred while generating prompt `{prompt}`: {e}")
+        await ctx.send(f"An error occurred while generating images: {e}")
 
 
 # UTILITY USER_CMD ------------------------------------------------------ END