Update main.py

main.py

@@ -107,7 +107,7 @@ app = FastAPI(
 )
 
 trans = SyncTranslator()
-BingImages = BingClient()
+BingImage = BingClient()
 
 timeout = 100
 contact_support = """
@@ -472,7 +472,8 @@ def gemini_pro(item: GeminiPro):
 
 @app.post("/Bing/Dall-E-3")
 async def bing_dalle(item: BingDalle):
-    try:
+    BingImages = await BingImage(image_provider=BingCreateImages)
+    try:        
         set_cookies(
             ".bing.com",
             {
@@ -482,22 +483,19 @@ async def bing_dalle(item: BingDalle):
     except requests.exceptions.RequestException:
         raise HTTPException(status_code=500, detail="Invalid cookie string, check your cookie string and try again")
 
-    async def generate_image(model, prompt):
+    async def generate_image(prompt, model):
         try:
-            response = await BingCreateImages().create_async(prompt=prompt)
-
-            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"}
+            response = await BingImages.images.generate(
+                prompt=prompt,
+                model=model,
+            )
         except BaseException as e:
             return {"status": "false", "message": f"Internal Error: {e}"}
-
-    try:
-        response = await generate_image(item.model, item.prompt)
         return response
+    try:
+        response = await generate_image(item.prompt, item.model)
+        image_url = response.data[0].url
+        return {"status": "true", "sukuna": {"message": image_url}}
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}