Update main.py

main.py

@@ -472,12 +472,7 @@ def gemini_pro(item: GeminiPro):
 
 @app.post("/Bing/Dall-E-3")
 async def bing_dalle(item: BingDalle):
-    try:
-        BingImages = await BingClient(image_provider=BingCreateImages)
-    except Exception:
-        print("Not awaited")
-        BingImages = BingClient(image_provider=BingCreateImages)
-    
+    BingImages = BingClient()
     try:        
         set_cookies(
             ".bing.com",
@@ -491,21 +486,20 @@ async def bing_dalle(item: BingDalle):
     async def generate_image(prompt, model):
         try:
             response = await BingImages.images.generate(
-                prompt=BingCreateImages.create(prompt),
+                prompt=prompt,
                 model=model,
             )
+
+            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"}
         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
-        if image_url is not None:
-            return SuccessResponse(status="True", randydev={"message": image_url})
-        else:
-            return {"status": "false", "message": f"No image URL found in the {response}"}
+        return response
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}