Update app.py

app.py

@@ -1,7 +1,127 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 
 app = FastAPI()
 
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
+
+@app.post("/bingimage")
+async def bing_image(request: Request):
+    data = await request.json()
+    prompt = data.get("prompt")
+    # Placeholder for actual data to return
+    import asyncio
+    import json
+    import re
+    from urllib.parse import quote
+    gist_url = "https://gist.githubusercontent.com/parasshah10/1519bed2f7f00e399f4a6b9300e84fc1/raw"
+    import httpx
+    
+    TIMEOUT = 200
+    TOKEN_FILE = "token.json"
+    BASE_URL = "https://www.bing.com"
+    USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36 Edg/120.0.0.0"
+    
+    class BingDalle:
+        def __init__(self, auth_cookie):
+            self.auth_cookie = self._load_config(auth_cookie)
+            self.headers = {
+                "User-Agent": USER_AGENT,
+                "Cookie": f"_U={self.auth_cookie}",
+            }
+            self.client = httpx.AsyncClient(
+                base_url=BASE_URL, headers=self.headers, timeout=TIMEOUT
+            )
+    
+        async def __aenter__(self):
+            return self
+    
+        async def __aexit__(self, *_):
+            await self.client.aclose()
+    
+        def _load_config(self, auth_cookie):
+            if auth_cookie:
+                return auth_cookie
+            try:
+                with open(TOKEN_FILE, "r") as file:
+                    config = json.load(file)
+                return config.get("_U")
+            except FileNotFoundError:
+                raise ValueError("Auth cookie not provided and token file not found.")
+    
+        async def _get_coins(self):
+            response = await self.client.get("/images/create")
+            if response.is_success:
+                coins_match = re.search(r'coins available">(\d+)<', response.text)
+                return coins_match[1] if coins_match else None
+            return None
+    
+        async def _poll_results(self, prompt, request_id):
+            encoded_prompt = quote(prompt)
+            result_url = f"/images/create/async/results/{request_id}?q={encoded_prompt}"
+            while True:
+                response = await self.client.get(result_url)
+                if response.is_success and "gir_async" in response.text:
+                    return response
+                await asyncio.sleep(5)
+    
+        def _construct_url(self, coins, prompt):
+            encoded_prompt = quote(prompt)
+            rt_value = "4" if int(coins) > 0 else "3"
+            return f"/images/create?q={encoded_prompt}&rt={rt_value}&FORM=GENCRE"
+    
+        async def _get_request_id(self, prompt, post_url):
+            data = {"q": prompt, "qs": "ds"}
+            response = await self.client.post(post_url, data=data, follow_redirects=True)
+            if response.is_success:
+                request_id = re.search(r"id=([^&]+)", str(response.url))
+                return request_id[1] if request_id else None
+            return None
+    
+        def _handle_poll_result(self, poll_results, prompt):
+            src_urls = list(
+                {
+                    url.split("?w=")[0]
+                    for url in re.findall(r'src="([^"]+)"', poll_results.text)
+                    if url.startswith(("http", "https")) and url.endswith("ImgGn")
+                }
+            )
+            return [{"url": src_url} for src_url in src_urls]
+    
+        async def generate_images(self, prompt):
+            coins = await self._get_coins()
+            post_url = self._construct_url(coins, prompt)
+            request_id = await self._get_request_id(prompt, post_url)
+            poll_results = await self._poll_results(prompt, request_id)
+            return self._handle_poll_result(poll_results, prompt)
+    
+    async def upload_to_imagebb(image_url, api_key):
+        upload_url = "https://api.imgbb.com/1/upload"
+        params = {
+            "key": api_key,
+            "image": image_url,
+        }
+        async with httpx.AsyncClient() as client:
+            response = await client.post(upload_url, params=params)
+            if response.is_success:
+                return response.json()["data"]["url"]
+            return None
+    
+    async def main(prompt, auth_cookie, imagebb_api_key):
+        async with BingDalle(auth_cookie) as bing:
+            image_urls = await bing.generate_images(prompt)
+            uploaded_urls = []
+            for image in image_urls:
+                uploaded_url = await upload_to_imagebb(image["url"], imagebb_api_key)
+                if uploaded_url:
+                    uploaded_urls.append(uploaded_url)
+            return uploaded_urls
+    
+    # Usage example
+    auth_cookie = requests.get(gist_url).text
+    imagebb_api_key = "769cca89ac6111406b57de8a4fc1e613"
+    prompt = "A cute dog"
+    
+    uploaded_urls = asyncio.run(main(prompt, auth_cookie, imagebb_api_key))
+    return {"images": uploaded_urls}