hendrik
/
matrix-pygmalion-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

exception handling

master
Hendrik Langer 2 years ago
parent
32021270ba
commit
ed61f5f67e
2 changed files with 32 additions and 26 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 4
      matrix_pygmalion_bot/ai/runpod_pygmalion.py
  2. 54
      matrix_pygmalion_bot/core.py

4
matrix_pygmalion_bot/ai/runpod_pygmalion.py
View File

@ -196,7 +196,7 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str,
# Make the request
r = requests.post(endpoint, json=input_data, headers=headers)
r_json = r.json()
logger.info(r_json)
logger.debug(r_json)
if r.status_code == 200:
status = r_json["status"]
@ -210,7 +210,7 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str,
endpoint = api_url + "status/" + job_id
r = requests.get(endpoint, headers=headers)
r_json = r.json()
logger.info(r_json)
logger.debug(r_json)
status = r_json["status"]
if status == 'IN_PROGRESS':
await asyncio.sleep(DELAY)

54
matrix_pygmalion_bot/core.py
View File

@ -107,31 +107,37 @@ class Callbacks(object):
negative_prompt = "ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, signature, cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face"
# else:
# negative_prompt = "ugly, deformed, out of frame"
if self.bot.service == "runpod":
if num == 1:
output = await ai.generate_image1(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 2:
output = await ai.generate_image2(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 3:
output = await ai.generate_image3(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 4:
output = await ai.generate_image4(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 5:
output = await ai.generate_image5(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 6:
output = await ai.generate_image6(prompt, negative_prompt, self.bot.runpod_api_key)
try:
if self.bot.service == "runpod":
if num == 1:
output = await ai.generate_image1(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 2:
output = await ai.generate_image2(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 3:
output = await ai.generate_image3(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 4:
output = await ai.generate_image4(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 5:
output = await ai.generate_image5(prompt, negative_prompt, self.bot.runpod_api_key)
elif num == 6:
output = await ai.generate_image6(prompt, negative_prompt, self.bot.runpod_api_key)
else:
raise ValueError('no image generator with that number')
elif self.bot.service == "stablehorde":
if num == 1:
output = await ai.generate_image1(prompt, negative_prompt, self.bot.stablehorde_api_key)
elif num == 2:
output = await ai.generate_image2(prompt, negative_prompt, self.bot.stablehorde_api_key)
elif num == 3:
output = await ai.generate_image3(prompt, negative_prompt, self.bot.stablehorde_api_key)
else:
raise ValueError('no image generator with that number')
else:
return
elif self.bot.service == "stablehorde":
if num == 1:
output = await ai.generate_image1(prompt, negative_prompt, self.bot.stablehorde_api_key)
elif num == 2:
output = await ai.generate_image2(prompt, negative_prompt, self.bot.stablehorde_api_key)
elif num == 3:
output = await ai.generate_image3(prompt, negative_prompt, self.bot.stablehorde_api_key)
else:
return
else:
raise ValueError('remote image generation not configured properly')
except ValueError as err:
errormessage = f"<ERROR> {err=}, {type(err)=}"
logger.error(errormessage)
await self.bot.send_message(self.client, room.room_id, errormessage)
return
for imagefile in output:

Write Preview
Loading…
Cancel
Save