From 22354ec275af65fbfe6cd72076bfe6cf7d227b5b Mon Sep 17 00:00:00 2001 From: phixxy Date: Sun, 21 Jan 2024 13:57:21 -0800 Subject: [PATCH] fixed bugs in reimagine if prompt missing. removed unused prompt code --- extensions/stable_diffusion.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/extensions/stable_diffusion.py b/extensions/stable_diffusion.py index 9ec7329..71d736f 100644 --- a/extensions/stable_diffusion.py +++ b/extensions/stable_diffusion.py @@ -329,21 +329,18 @@ class StableDiffusion(commands.Cog): try: if ctx.message.attachments: file_url = ctx.message.attachments[0].url - prompt = ctx.message.content.split(" ", maxsplit=1)[1] elif ctx.message.content.startswith("!reimagine "): file_url = ctx.message.content.split(" ", maxsplit=2)[1] - try: - prompt = ctx.message.content.split(" ", maxsplit=2)[2] - except: - prompt = "" else: - print("No image linked or attached.") + await ctx.send("No image linked or attached.") return except Exception as error: await self.handle_error(error) print("Couldn't find image.") return prompt = self.get_prompt_from_ctx(ctx) + if not prompt: + prompt = "" key_value_pairs = self.get_kv_from_ctx(ctx) try: async with self.bot.http_session.get(file_url) as response: