exception handling

matrix_pygmalion_bot/ai/runpod_pygmalion.py

@@ -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)

matrix_pygmalion_bot/core.py

@@ -107,6 +107,7 @@ 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"
+            try:
                 if self.bot.service == "runpod":
                     if num == 1:
                         output = await ai.generate_image1(prompt, negative_prompt, self.bot.runpod_api_key)
@@ -121,7 +122,7 @@ class Callbacks(object):
                     elif num == 6:
                         output = await ai.generate_image6(prompt, negative_prompt, self.bot.runpod_api_key)
                     else:
-                    return
+                        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)
@@ -130,8 +131,13 @@ class Callbacks(object):
                     elif num == 3:
                         output = await ai.generate_image3(prompt, negative_prompt, self.bot.stablehorde_api_key)
                     else:
-                    return
+                        raise ValueError('no image generator with that number')
                 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: