Update main.py

main.py

@@ -476,6 +476,7 @@ async def bing_dalle(item: BingDalle):
         BingImages = await BingClient(image_provider=BingCreateImages)
     except Exception:
         pass
+    
     try:        
         set_cookies(
             ".bing.com",
@@ -494,11 +495,19 @@ async def bing_dalle(item: BingDalle):
             )
         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}}
+        
+        # Check if response contains the expected data
+        if 'data' in response and response['data']:
+            # Assuming the URL is stored in 'url' key of the first item
+            image_url = response['data'][0]['url']
+            return {"status": "true", "sukuna": {"message": image_url}}
+        else:
+            return {"status": "false", "message": "No image URL found in the response"}
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}