diff --git a/weeeabot.py b/weeeabot.py index 6dd2165..195df19 100644 --- a/weeeabot.py +++ b/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,23 +70,29 @@ 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") - 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() + user_roles = [role.name for role in ctx.message.author.roles] - 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) + 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: + 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() @@ -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)