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
Unified 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 # Make the request
r = requests.post(endpoint, json=input_data, headers=headers) r = requests.post(endpoint, json=input_data, headers=headers)
r_json = r.json() r_json = r.json()
logger.info(r_json) logger.debug(r_json)
if r.status_code == 200: if r.status_code == 200:
status = r_json["status"] 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 endpoint = api_url + "status/" + job_id
r = requests.get(endpoint, headers=headers) r = requests.get(endpoint, headers=headers)
r_json = r.json() r_json = r.json()
logger.info(r_json) logger.debug(r_json)
status = r_json["status"] status = r_json["status"]
if status == 'IN_PROGRESS': if status == 'IN_PROGRESS':
await asyncio.sleep(DELAY) 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" 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: # else:
# negative_prompt = "ugly, deformed, out of frame" # negative_prompt = "ugly, deformed, out of frame"
if self.bot.service == "runpod": try:
if num == 1: if self.bot.service == "runpod":
output = await ai.generate_image1(prompt, negative_prompt, self.bot.runpod_api_key) if num == 1:
elif num == 2: output = await ai.generate_image1(prompt, negative_prompt, self.bot.runpod_api_key)
output = await ai.generate_image2(prompt, negative_prompt, self.bot.runpod_api_key) elif num == 2:
elif num == 3: output = await ai.generate_image2(prompt, negative_prompt, self.bot.runpod_api_key)
output = await ai.generate_image3(prompt, negative_prompt, self.bot.runpod_api_key) elif num == 3:
elif num == 4: output = await ai.generate_image3(prompt, negative_prompt, self.bot.runpod_api_key)
output = await ai.generate_image4(prompt, negative_prompt, self.bot.runpod_api_key) elif num == 4:
elif num == 5: output = await ai.generate_image4(prompt, negative_prompt, self.bot.runpod_api_key)
output = await ai.generate_image5(prompt, negative_prompt, self.bot.runpod_api_key) elif num == 5:
elif num == 6: output = await ai.generate_image5(prompt, negative_prompt, self.bot.runpod_api_key)
output = await ai.generate_image6(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: else:
return raise ValueError('remote image generation not configured properly')
elif self.bot.service == "stablehorde": except ValueError as err:
if num == 1: errormessage = f"<ERROR> {err=}, {type(err)=}"
output = await ai.generate_image1(prompt, negative_prompt, self.bot.stablehorde_api_key) logger.error(errormessage)
elif num == 2: await self.bot.send_message(self.client, room.room_id, errormessage)
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:
return return
for imagefile in output: for imagefile in output:

Write Preview
Loading…
Cancel
Save