Craiyon API Image generator PR

bot/commands/fun/gen.py

@@ -0,0 +1,33 @@
+from io import BytesIO
+import base64
+import time
+from craiyon import Craiyon
+from bot.commands.fun.__blocked_prompts import blocked
+from bot.config import Config, Embed
+from bot.base import Command
+
+
+class cmd(Command):
+    """ A discord command instance. """
+# 
+    name = "gen"
+    usage = "gen <*prompt>"
+    description = "Uses the Craiyon AI API to generate an image with a given prompt"
+
+    async def execute(self, arguments, message) -> None:
+        prompt = arguments[0]
+        if prompt.lower() in blocked:
+            await message.channel.send("I'm not generating that you meanie >:( It's NSFW! Try something else!")
+            return
+
+        loadingE = Embed(title=f"Image {prompt} currently generating, please wait a moment!")
+        loadingM = await message.channel.send(embed=loadingE)
+
+        generator = Craiyon()
+        result = generator.generate(prompt)
+        images = result.images
+        embed = Embed(title="Here is the generated image")
+        embed.set_image(url=images[0])
+
+        await loadingM.delete()
+        await message.channel.send(embed=embed)

bot/events/blocked_prompts.py

@@ -0,0 +1,25 @@
+blocked = [
+    "nsfw",
+    "pussy",
+    "gore",
+    "murder",
+    "penis",
+    "cat",
+    "emacs",
+    "porn",
+    "vagina",
+    "dick",
+    "sex",
+    "intercourse",
+    "bondage",
+    "bdsm",
+    "fetish",
+    "kink",
+    "pegging",
+    "peg",
+    "anal",
+    "vaginal",
+    "erotic",
+    "notepad
+
+]