Update main.py

main.py

@@ -487,13 +487,19 @@ async def bing_dalle(item: BingDalle):
                 prompt=prompt,
                 model=model,
             )
+
+            image_url = response.get('data', [])[0].get('url')
+            return {"status": "true", "sukuna": {"message": image_url}}
+        except BingImagesException as e:
+            return {"status": "false", "message": f"BingImages Error: {e}"}
+        except IndexError:
+            return {"status": "false", "message": "No image URL found in the response"}
         except BaseException as e:
             return {"status": "false", "message": f"Internal Error: {e}"}
-        return response
+
     try:
         response = await generate_image(item.prompt, item.model)
-        image_url = response.data[0].url
-        return {"status": "true", "sukuna": {"message": image_url}}
+        return response
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}