weeeabot.py

@@ -17,7 +17,7 @@ except FileNotFoundError:
 intents = discord.Intents.default()
 intents.message_content = True
 
-api = diffuseapi.DiffuseAPI("https://art.jurydoak.com", ["Bot"], True, 28)
+api = diffuseapi.DiffuseAPI("https://art.jurydoak.com", ["Bot"], False, 28)
 
 # client = discord.Client(intents=intents)
 
@@ -70,29 +70,23 @@ bot = commands.Bot(intents=intents, command_prefix="!", activity=activity, help_
 async def prompt(ctx, *args):
     '''Generate an image with the provided prompt'''
     prompt = " ".join(args)
-    user_roles = [role.name for role in ctx.message.author.roles]
+    await ctx.reply("Generating your image boo")
+    image_data = await api.generate_image(prompt)
+    view = Confirm()
+    if image_data is None:
+        await ctx.reply("Something went wrong, please report this to the admin so it can be ignored")
+        return
+    decoded_image = b64decode(image_data)
+    h = hashlib.md5()
+    h.update(decoded_image)
+    digest = h.digest()
 
-    async with ctx.typing():
+    if digest == b'i\xac`\xde\xbak\xba\xab{2Z\xcc\tK\xc2~':
-        if "Bot Ban" not in user_roles:
+        await ctx.reply("Were no stranger to lewds, but you know the rules, and so do I", file=discord.File("images/404.jpg", "404.jpg"))
-            # await ctx.reply("Generating your image boo")
+        return
-            image_data = await api.generate_image(prompt)
+    print(h.digest())
-            view = Confirm()
+    data = io.BytesIO(decoded_image)
-            if image_data is None:
+    await ctx.reply("", file=discord.File(data, "_".join(args) + ".png"), view=view)
-                await ctx.reply("Something went wrong, please report this to the admin so it can be ignored")
-                return
-            decoded_image = b64decode(image_data)
-            h = hashlib.md5()
-            h.update(decoded_image)
-            digest = h.digest()
-
-            if digest == b'i\xac`\xde\xbak\xba\xab{2Z\xcc\tK\xc2~':
-                await ctx.reply("Were no stranger to lewds, but you know the rules, and so do I", file=discord.File("images/404.jpg", "404.jpg"))
-                return
-            print(h.digest())
-            data = io.BytesIO(decoded_image)
-            await ctx.reply("", file=discord.File(data, "_".join(args) + ".png"), view=view)
-        else:
-            await ctx.reply("NO!")
 
 
 @bot.command()
@@ -121,7 +115,7 @@ async def steps(ctx, arg):
         return
 
     # global num_steps
-    api.num_steps = arg
+    api.set_steps(arg)
 
     await ctx.reply(f"I have updated the steps to {api.num_steps} for you my master.")
 
@@ -132,7 +126,7 @@ async def settings(ctx):
     settings = f"""
     ```
     seed: {api.seed}
-    steps: {api.num_steps}
+    steps: {api.steps}
     ```
     """
     await ctx.message.channel.send(settings)