Merge pull request 'enable pr0n' (#3) from stable-diffuse-api into master

Reviewed-on: #3

weeeabot.py

@@ -17,7 +17,7 @@ except FileNotFoundError:
 intents = discord.Intents.default()
 intents.message_content = True
 
-api = diffuseapi.DiffuseAPI("https://art.jurydoak.com", ["Bot"], False, 28)
+api = diffuseapi.DiffuseAPI("https://art.jurydoak.com", ["Bot"], True, 28)
 
 # client = discord.Client(intents=intents)
 
@@ -70,7 +70,11 @@ 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)
-    await ctx.reply("Generating your image boo")
+    user_roles = [role.name for role in ctx.message.author.roles]
+
+    async with ctx.typing():
+        if "Bot Ban" not in user_roles:
+            # await ctx.reply("Generating your image boo")
             image_data = await api.generate_image(prompt)
             view = Confirm()
             if image_data is None:
@@ -87,6 +91,8 @@ async def prompt(ctx, *args):
             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()
@@ -115,7 +121,7 @@ async def steps(ctx, arg):
         return
 
     # global num_steps
-    api.set_steps(arg)
+    api.num_steps = arg
 
     await ctx.reply(f"I have updated the steps to {api.num_steps} for you my master.")
 
@@ -126,7 +132,7 @@ async def settings(ctx):
     settings = f"""
     ```
     seed: {api.seed}
-    steps: {api.steps}
+    steps: {api.num_steps}
     ```
     """
     await ctx.message.channel.send(settings)