Bing Dall-E-3 (#11)

- Update main.py (5f87b05bcce586ddc8cf21c35d9cdbf5367a761e)


Co-authored-by: True Saiyan <[email protected]>

main.py

@@ -58,6 +58,11 @@ from fastapi import Body, Query
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 
+import g4f
+from g4f.client import Client as BingClient
+from g4f.cookies import set_cookies
+from g4f.Provider import BingCreateImages, OpenaiChat, Gemini
+
 from pymongo import MongoClient
 
 from RyuzakiLib.hackertools.chatgpt import RendyDevChat
@@ -537,6 +542,31 @@ def ryuzaki_ai(
     except Exception:
         return SuccessResponse(status="False", randydev={"message": contact_support})
 
+
+BingImages = BingClient()
+
+@app.post("/ryuzaki/dall-e-3")
+def bing_dalle(item: BingDalle, api_key: None = Depends(validate_api_key)):
+    try:        
+        set_cookies(
+            ".bing.com",
+            {
+                "_U": item.cookie
+            },
+        )
+    except requests.exceptions.RequestException:
+        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 SuccessResponse(status="True", randydev={"data": response.data[0].url})
+
+    except BaseException as e:
+        return SuccessResponse(status="False", randydev={"data": f"Error: {e}"})
+
 @app.post("/ryuzaki/dalle3xl")
 def dalle_3xl(
     item: Dalle3XL,