update 1.3.1 added /image command to generate ai images

Dockerfile

@@ -2,7 +2,7 @@ FROM python:3.12-slim
 
 WORKDIR /code
 
-RUN pip install --no-cache-dir -U discord pydvpl fastapi uvicorn psutil python-dotenv python-tgpt
+RUN pip install --no-cache-dir -U discord.py pydvpl fastapi uvicorn psutil python-dotenv python-tgpt craiyon.py
 
 COPY ./pydvpl_bot.py /code
 

pydvpl_bot.py

@@ -15,6 +15,10 @@ 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
+
 
 
 # ENV_LOADER ------------------------------------------------------ START
@@ -29,7 +33,7 @@ load_dotenv()
 # DEFINES ------------------------------------------------------ START
 
 
-BOT_VERSION = '1.3.0'
+BOT_VERSION = '1.3.1'
 
 AUTHORIZED_USER_ID = os.getenv('AUTHORIZED_ID')
 
@@ -553,6 +557,36 @@ async def chat(ctx, model, *, message):
     await ctx.send(response)
 
 
+# FUNCTION ------------------------------------------------------ SPLIT
+
+
+@bot.hybrid_command(help="Generate images with 'craiyon' ai.")
+async def image(ctx, *, prompt: str):
+
+    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.")
+        return
+
+    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
+
+        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
+
+    except Exception as e:
+        await ctx.send(f"An error occurred while generating prompt `{prompt}`: {e}")
+
+
 # UTILITY USER_CMD ------------------------------------------------------ END