phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed bug if no prompt is given in reimagine

Browse source
This commit is contained in:
phixxy 2024-01-21 13:49:28 -08:00
parent 149c7591de
commit ef5b841b86
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
extensions/stable_diffusion.py
View file

@ -332,7 +332,10 @@ class StableDiffusion(commands.Cog):
prompt = ctx.message.content.split(" ", maxsplit=1)[1] prompt = ctx.message.content.split(" ", maxsplit=1)[1]
elif ctx.message.content.startswith("!reimagine "): elif ctx.message.content.startswith("!reimagine "):
file_url = ctx.message.content.split(" ", maxsplit=2)[1] 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: else:
print("No image linked or attached.") print("No image linked or attached.")
return return