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

removed handle_error

Browse source
This commit is contained in:
phixxy 2024-01-24 23:54:17 -08:00
parent 8491f02227
commit 7b5f7f630b
1 changed files with 7 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

25
extensions/stable_diffusion.py
View file

@ -46,16 +46,7 @@ class StableDiffusion(commands.Cog):
response = response_data['choices'][0]['message']['content'] response = response_data['choices'][0]['message']['content']
return response return response
except Exception as error: except Exception as error:
return await self.handle_error(error) return "Error in answer question in stable_diffusion"
async def handle_error(self, error): # This needs to be deleted and replaced with logging
print(error)
current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
log_line = current_time + ': ' + str(error) + '\n'
file_name = self.data_dir + "error_log.log"
with open(file_name, 'a') as f:
f.write(log_line)
return error
def get_kv_from_ctx(self, ctx): def get_kv_from_ctx(self, ctx):
try: try:
@ -114,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:
await self.handle_error(error) print("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
@ -235,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.")
await self.handle_error(error) print("Error in imagine 1")
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])))
@ -246,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.")
await self.handle_error(self, error) print("error in imagine 2")
pnginfo = PngImagePlugin.PngInfo() pnginfo = PngImagePlugin.PngInfo()
pnginfo.add_text("parameters", response2.get("info")) pnginfo.add_text("parameters", response2.get("info"))
@ -292,7 +283,6 @@ class StableDiffusion(commands.Cog):
print("No image linked or attached.") print("No image linked or attached.")
return return
except Exception as error: except Exception as error:
await self.handle_error(error)
print("Couldn't find image.") print("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:
@ -313,7 +303,7 @@ class StableDiffusion(commands.Cog):
print(r) print(r)
await ctx.send(r.get("caption")) await ctx.send(r.get("caption"))
except Exception as error: except Exception as error:
await self.handle_error(error) print("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(
@ -335,7 +325,6 @@ 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:
await self.handle_error(error)
print("Couldn't find image.") print("Couldn't find image.")
return return
prompt = self.get_prompt_from_ctx(ctx) prompt = self.get_prompt_from_ctx(ctx)
@ -355,7 +344,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.")
await self.handle_error(error) print("error in reimagine 1")
img_link = await self.my_open_img_file(imageName) img_link = await self.my_open_img_file(imageName)
@ -386,7 +375,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.")
await self.handle_error(error) print("error in reimagine 2")
@commands.command( @commands.command(
description="Negative Prompt", description="Negative Prompt",