Update main.py

main.py

@@ -25,6 +25,7 @@ import logging
 import os
 import random
 import re
+import base64
 from base64 import b64decode
 from base64 import b64decode as kc
 from datetime import datetime as dt
@@ -584,12 +585,10 @@ async def gemini_pro(item: GeminiPro):
                 status="False", randydev={"message": contact_support}
             )
 
-
 BingImages = BingClient()
 
-
 @app.post("/Bing/Dall-E-3")
-def bing_dalle(item: BingDalle):
+async def bing_dalle(item: BingDalle):
     try:        
         set_cookies(
             ".bing.com",
@@ -601,16 +600,21 @@ def bing_dalle(item: BingDalle):
         raise HTTPException(status_code=500, detail="Invalid cookie string, check your cookie string and try again")
 
     try:
-        response = BingImages.images.generate(
-            prompt=item.prompt,
-            model=item.model,
-        )
-        return {"status": "true", "sukuna": {"url": response.data[0].url}}
+        response = await imgclient.images.generate(prompt=query, model="dall-e-3")
+        paths = []
+        base64_images = []
+
+        for index, item in enumerate(response.data):
+            image_url = item.url
+            image_data = requests.get(image_url).content
+            base64_image = base64.b64encode(image_data).decode('utf-8')
+            base64_images.append(base64_image)
+
+        return {"status": "true", "sukuna": {"urls": [item.url for item in response.data], "base64_images": base64_images}}
 
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}
 
-
 @app.post("/UFoP/dalle3xl")
 def dalle_3xl(
     item: Dalle3XL,