Update main.py

main.py

@@ -472,7 +472,7 @@ def gemini_pro(item: GeminiPro):
 
 @app.post("/Bing/Dall-E-3")
 async def bing_dalle(item: BingDalle):
-    try:        
+    try:
         set_cookies(
             ".bing.com",
             {
@@ -484,16 +484,14 @@ async def bing_dalle(item: BingDalle):
 
     async def generate_image(model, prompt):
         try:
-            model=model
-            response = await BingCreateImages().create_async(
-#                model=model,
-                prompt=prompt,
-            )
+            response = await BingCreateImages().create_async(prompt=prompt)
 
-            image_url = response.get('data', [])[0].get('url')
-            return {"status": "true", "sukuna": {"message": image_url}}
-        except IndexError:
-            return {"status": "false", "message": "No image URL found in the response"}
+            image_data = response.get('data')
+            if image_data:
+                image_url = image_data[0].get('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"Internal Error: {e}"}
 
@@ -504,7 +502,6 @@ async def bing_dalle(item: BingDalle):
         return {"status": "false", "message": f"Something went wrong: {e}"}
 
 
-
 @app.post("/UFoP/dalle3xl")
 def dalle_3xl(item: Dalle3XL, api_key: None = Depends(validate_api_key)):
     API_URL = SOURCE_DALLE3XL_URL