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

added better error handling when failure to connect to server

Browse source
This commit is contained in:
Phixxy 2024-02-11 14:34:28 -08:00
parent e906b9da1d
commit b421166b23
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
extensions/stable_diffusion.py
View file

@ -226,7 +226,8 @@ class StableDiffusion(commands.Cog):
r = await resp.json() r = await resp.json()
except Exception as error: except Exception as error:
await ctx.send("My image generation service may not be running.") await ctx.send("My image generation service may not be running.")
self.bot.logger.exception("Error in imagine") self.bot.logger.exception("Error with connection to image generation service")
return
for i in r['images']: for i in r['images']:
image = Image.open(io.BytesIO(base64.b64decode(i.split(",", 1)[0]))) image = Image.open(io.BytesIO(base64.b64decode(i.split(",", 1)[0])))