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

added logging

Browse source
This commit is contained in:
phixxy 2024-01-25 02:42:23 -08:00
parent 820c4fa8c6
commit c03393c818
1 changed files with 16 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

33
extensions/stable_diffusion.py
View file

@ -28,7 +28,7 @@ class StableDiffusion(commands.Cog):
if not os.path.exists(self.data_dir): if not os.path.exists(self.data_dir):
os.mkdir(self.data_dir) os.mkdir(self.data_dir)
except: except:
print("StableDiffusion failed to make directories") self.bot.logger.exception("StableDiffusion failed to make directories")
async def answer_question(self, topic, model="gpt-3.5-turbo"): # Only needed for draw command async def answer_question(self, topic, model="gpt-3.5-turbo"): # Only needed for draw command
headers = { headers = {
@ -83,12 +83,12 @@ class StableDiffusion(commands.Cog):
return return
for attachment in ctx.attachments: for attachment in ctx.attachments:
if attachment.url.endswith(('.jpg', '.png')): if attachment.url.endswith(('.jpg', '.png')):
print("image seen") self.bot.logger.debug("image seen")
async with self.bot.http_session.get(attachment.url) as response: async with self.bot.http_session.get(attachment.url) as response:
imageName = self.working_dir + str(time.time_ns()) + '.png' imageName = self.working_dir + str(time.time_ns()) + '.png'
with open(imageName, 'wb') as out_file: with open(imageName, 'wb') as out_file:
print('Saving image: ' + imageName) self.bot.logger.debug('Saving image: ' + imageName)
while True: while True:
chunk = await response.content.read(1024) chunk = await response.content.read(1024)
if not chunk: if not chunk:
@ -105,7 +105,7 @@ class StableDiffusion(commands.Cog):
description = description.split(',')[0] description = description.split(',')[0]
metadata += f"<image:{description}>\n" metadata += f"<image:{description}>\n"
except self.bot.aiohttp.ClientError as error: except self.bot.aiohttp.ClientError as error:
print("ERROR: CLIP may not be running. Could not look at image.") self.bot.logger.exception("ERROR: CLIP may not be running. Could not look at image.")
return "ERROR: CLIP may not be running. Could not look at image." return "ERROR: CLIP may not be running. Could not look at image."
return metadata return metadata
@ -226,7 +226,7 @@ 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.")
print("Error in imagine 1") self.bot.logger.exception("Error in imagine")
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])))
@ -237,7 +237,7 @@ class StableDiffusion(commands.Cog):
response2 = await resp.json() response2 = 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.")
print("error in imagine 2") self.bot.logger.exception("error in imagine")
pnginfo = PngImagePlugin.PngInfo() pnginfo = PngImagePlugin.PngInfo()
pnginfo.add_text("parameters", response2.get("info")) pnginfo.add_text("parameters", response2.get("info"))
@ -280,15 +280,15 @@ class StableDiffusion(commands.Cog):
elif ctx.message.attachments: elif ctx.message.attachments:
file_url = ctx.message.attachments[0].url file_url = ctx.message.attachments[0].url
else: else:
print("No image linked or attached.") self.bot.logger.debug("No image linked or attached.")
return return
except Exception as error: except Exception as error:
print("Couldn't find image.") self.bot.logger.exception("Couldn't find image.")
return return
async with self.bot.http_session.get(file_url) as response: async with self.bot.http_session.get(file_url) as response:
imageName = self.working_dir + str(time.time_ns()) + ".png" imageName = self.working_dir + str(time.time_ns()) + ".png"
with open(imageName, 'wb') as out_file: with open(imageName, 'wb') as out_file:
print(f"Saving image: {imageName}") self.bot.logger.debug(f"Saving image: {imageName}")
while True: while True:
chunk = await response.content.read(1024) chunk = await response.content.read(1024)
if not chunk: if not chunk:
@ -300,10 +300,9 @@ class StableDiffusion(commands.Cog):
payload = {"image": img_link} payload = {"image": img_link}
async with self.bot.http_session.post(url, json=payload) as response: async with self.bot.http_session.post(url, json=payload) as response:
r = await response.json() r = await response.json()
print(r)
await ctx.send(r.get("caption")) await ctx.send(r.get("caption"))
except Exception as error: except Exception as error:
print("error in describe") self.bot.logger.exception("error in describe")
await ctx.send("My image generation service may not be running.") await ctx.send("My image generation service may not be running.")
@commands.command( @commands.command(
@ -325,7 +324,7 @@ class StableDiffusion(commands.Cog):
await ctx.send("No image linked or attached.") await ctx.send("No image linked or attached.")
return return
except Exception as error: except Exception as error:
print("Couldn't find image.") self.bot.logger.exception("Couldn't find image.")
return return
prompt = self.get_prompt_from_ctx(ctx) prompt = self.get_prompt_from_ctx(ctx)
if not prompt: if not prompt:
@ -335,7 +334,7 @@ class StableDiffusion(commands.Cog):
async with self.bot.http_session.get(file_url) as response: async with self.bot.http_session.get(file_url) as response:
imageName = self.working_dir + str(time.time_ns()) + ".png" imageName = self.working_dir + str(time.time_ns()) + ".png"
with open(imageName, 'wb') as out_file: with open(imageName, 'wb') as out_file:
print(f"Saving image: {imageName}") self.bot.logger.debug(f"Saving image: {imageName}")
while True: while True:
chunk = await response.content.read(1024) chunk = await response.content.read(1024)
if not chunk: if not chunk:
@ -344,7 +343,7 @@ class StableDiffusion(commands.Cog):
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.")
print("error in reimagine 1") self.bot.logger.exception("error in reimagine 1")
img_link = await self.my_open_img_file(imageName) img_link = await self.my_open_img_file(imageName)
@ -375,7 +374,7 @@ class StableDiffusion(commands.Cog):
await ctx.send(file=f) await ctx.send(file=f)
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.")
print("error in reimagine 2") self.bot.logger.exception("error in reimagine 2")
@commands.command( @commands.command(
description="Negative Prompt", description="Negative Prompt",
@ -395,7 +394,7 @@ class StableDiffusion(commands.Cog):
async def setup(bot): async def setup(bot):
try: try:
await bot.add_cog(StableDiffusion(bot)) await bot.add_cog(StableDiffusion(bot))
print("Successfully added StableDiffusion Cog") bot.logger.info("Successfully added StableDiffusion Cog")
except: except:
print("Failed to load StableDiffusion Cog") bot.logger.info("Failed to load StableDiffusion Cog")