Update main.py

main.py

@@ -472,7 +472,6 @@ def gemini_pro(item: GeminiPro):
 
 @app.post("/Bing/Dall-E-3")
 async def bing_dalle(item: BingDalle):
-    BingImages = BingClient()
     try:        
         set_cookies(
             ".bing.com",
@@ -483,23 +482,13 @@ 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(prompt, model):
-        try:
-            response = await BingImages.images.generate(
-                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}"}
-
     try:
-        response = await generate_image(item.prompt, item.model)
-        return response
+        response = BingImages.images.generate(
+            prompt=item.prompt,
+            model=item.model,
+        )
+        return {"status": "true", "sukuna": {"message": response.data[0].url}}
+
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}