From ef5b841b8689444ed7d57bfc352fa35aaaff0609 Mon Sep 17 00:00:00 2001 From: phixxy Date: Sun, 21 Jan 2024 13:49:28 -0800 Subject: [PATCH] fixed bug if no prompt is given in reimagine --- extensions/stable_diffusion.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/extensions/stable_diffusion.py b/extensions/stable_diffusion.py index 8bf6e18..9ec7329 100644 --- a/extensions/stable_diffusion.py +++ b/extensions/stable_diffusion.py @@ -332,7 +332,10 @@ class StableDiffusion(commands.Cog): prompt = ctx.message.content.split(" ", maxsplit=1)[1] elif ctx.message.content.startswith("!reimagine "): file_url = ctx.message.content.split(" ", maxsplit=2)[1] - prompt = ctx.message.content.split(" ", maxsplit=2)[2] + try: + prompt = ctx.message.content.split(" ", maxsplit=2)[2] + except: + prompt = "" else: print("No image linked or attached.") return