Update main.py

main.py

@@ -36,6 +36,10 @@ REQUEST_TIMEOUT = 300
 
 # ─── Headers ───
 HEADERS = {
+    "authority": "www.blackbox.ai",
+    "method": "POST",
+    "path": "/api/chat",
+    "scheme": "https",
     "accept": "*/*",
     "accept-encoding": "gzip, deflate, br, zstd",
     "accept-language": "en-US,en;q=0.9",
@@ -84,14 +88,10 @@ def random_id(): return _rand(21, string.digits)
 def random_customer_id(): return "cus_" + _rand(12)
 
 def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
-    unique_id = str(uuid.uuid4())
-    msg_id = _rand(8)
-    if messages and isinstance(messages[-1], dict):
-        msg_id = messages[-1].get("id", msg_id)
     return {
         "messages": messages,
         "agentMode": {},
-        "id": msg_id,
+        "id": None,
         "previewToken": None,
         "userId": None,
         "codeModelMode": True,
@@ -110,7 +110,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "isMemoryEnabled": False,
         "mobileClient": False,
         "userSelectedModel": None,
-        "validated": unique_id,
+        "validated": str(uuid.uuid4()),
         "imageGenerationMode": False,
         "webSearchModePrompt": False,
         "deepSearchMode": True,
@@ -126,20 +126,20 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         },
         "session": {
             "user": {
-                "name": _rand(10),
-                "email": random_email(),
-                "image": "https://lh3.googleusercontent.com/a/default",
-                "id": random_id()
+                "name": "S.C gaming",
+                "email": "[email protected]",
+                "image": "https://lh3.googleusercontent.com/a/ACg8ocI-ze5Qe42S-j8xaCL6X7KSVwfiOae4fONqpTxzt0d2_a2FIld1=s96-c",
+                "id": "100846841133312010974"
             },
-            "expires": "2025-12-31T23:59:59Z",
+            "expires": "2025-06-09T19:36:08.220Z",
             "isNewUser": False
         },
         "isPremium": True,
         "subscriptionCache": {
             "status": "PREMIUM",
-            "customerId": random_customer_id(),
-            "expiryTimestamp": int(time.time()) + 30 * 86400,
-            "lastChecked": int(time.time()),
+            "customerId": "cus_Rtiok41PQ2oo1c",
+            "expiryTimestamp": 1749108685,
+            "lastChecked": 1746822333827,
             "isTrialSubscription": False
         },
         "beastMode": False,
@@ -185,9 +185,9 @@ async def get_blackbox_response(*, data, stream: bool, request_id: str) -> Async
                         yield chunk.decode("utf-8", "ignore")
             else:
                 yield await resp.text()
-    except aiohttp.ContentEncodingError as e:
-        logger.error("[%s] Brotli decode failed: %s", request_id, e)
-        raise HTTPException(status_code=502, detail="Install Brotli for br-encoded content") from e
+    except Exception as e:
+        logger.exception("[%s] Unexpected upstream error", request_id)
+        raise HTTPException(status_code=502, detail="Failed to contact upstream") from e
 
 @app.middleware("http")
 async def add_request_id(request: Request, call_next):