detect content error from dalle3

bghira · Apr 18, 2024 · d779c3b · d779c3b
1 parent dc7825b
commit d779c3b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/discord_tron_master/classes/openai/text.py b/discord_tron_master/classes/openai/text.py
@@ -188,10 +188,14 @@ async def dalle_image_generate(self, prompt, user_config: dict):
             if "error" in response:
                 logging.error(f"Error generating image: {response}")
                 raise Exception(f'{response["error"]["code"]}: {response["error"]["message"]}')
+            else:
+                logging.debug(f"Received response from OpenAI image endpoint: {response}")
 
             url = response.data[0].url
+            logging.debug(f"Retrieving URL: {url}")
             # retrieve URL, return Image
             image_obj = await self.retrieve_image(url)
+            logging.debug(f"Result: {image_obj}")
             if not hasattr(image_obj, "size"):
                 logging.error(f"Image object does not have a size attribute. Returning None.")
                 logging.debug(f"Response from OpenAI: {response}")