update test

pydvpl_bot.py

@@ -15,10 +15,9 @@ 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
 
 
 # ENV_LOADER ------------------------------------------------------ START
@@ -569,20 +568,22 @@ async def image(ctx, *, prompt: str):
 
     await ctx.send(f"Generating 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
 
-    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
-
-    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
+        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
 
-    await ctx.reply(files=images1)  # Reply to the user with all 9 images in 1 message
+        await ctx.reply(files=images1)  # Reply to the user with all 9 images in 1 message
 
+    except Exception as e:
+        await ctx.send(f"An error occurred while generating prompt `{prompt}`: {e}")
 
 
 # UTILITY USER_CMD ------------------------------------------------------ END