Update main.py

main.py

@@ -9,7 +9,6 @@ from typing import AsyncGenerator, Dict, List, Any
 
 import aiohttp
 import brotli
-import sys
 import importlib
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
@@ -110,7 +109,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "isMemoryEnabled": False,
         "mobileClient": False,
         "userSelectedModel": None,
-        "validated": str(uuid.uuid4()),
+        "validated": "00f37b34-a166-4efb-bce5-1312d87f2f94",
         "imageGenerationMode": False,
         "webSearchModePrompt": False,
         "deepSearchMode": True,
@@ -126,10 +125,10 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         },
         "session": {
             "user": {
-                "name": "S.C gaming",
-                "email": "simarmanbir@gmail.com",
-                "image": "https://lh3.googleusercontent.com/a/ACg8ocI-ze5Qe42S-j8xaCL6X7KSVwfiOae4fONqpTxzt0d2_a2FIld1=s96-c",
-                "id": "100846841133312010974"
+                "name": _rand(10),
+                "email": random_email(),
+                "image": "https://lh3.googleusercontent.com/a/default",
+                "id": random_id()
             },
             "expires": "2025-06-09T19:36:08.220Z",
             "isNewUser": False
@@ -137,7 +136,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "isPremium": True,
         "subscriptionCache": {
             "status": "PREMIUM",
-            "customerId": "cus_Rtiok41PQ2oo1c",
+            "customerId": random_customer_id(),
             "expiryTimestamp": 1749108685,
             "lastChecked": 1746822333827,
             "isTrialSubscription": False
@@ -147,6 +146,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "designerMode": False
     }
 
+# ─── Retry Logic ───
 def log_retry(retry_state):
     rid = retry_state.kwargs.get("request_id", "unknown")
     attempt = retry_state.attempt_number
@@ -189,6 +189,7 @@ async def get_blackbox_response(*, data, stream: bool, request_id: str) -> Async
         logger.exception("[%s] Unexpected upstream error", request_id)
         raise HTTPException(status_code=502, detail="Failed to contact upstream") from e
 
+# ─── Middleware ───
 @app.middleware("http")
 async def add_request_id(request: Request, call_next):
     request.state.request_id = rid = str(uuid.uuid4())
@@ -198,6 +199,7 @@ async def add_request_id(request: Request, call_next):
     logger.info("[%s] finished in %.2fs", rid, time.perf_counter() - start)
     return resp
 
+# ─── Routes ───
 @app.get("/")
 async def root():
     return {"message": "API is running"}
@@ -247,8 +249,12 @@ async def chat_completions(request: Request):
                     "model": "DeepResearch",
                     "choices": [{"index": 0, "delta": {"content": chunk}}],
                 }
-                yield f"data: {json.dumps(msg)}\n\n"
-            yield "data: [DONE]\n\n"
+                yield f"data: {json.dumps(msg)}
+
+"
+            yield "data: [DONE]
+
+"
 
         return StreamingResponse(event_stream(), media_type="text/event-stream")
 
@@ -259,4 +265,4 @@ async def chat_completions(request: Request):
         raise HTTPException(status_code=502, detail="Blackbox upstream failed")
     except Exception as e:
         logger.exception("[%s] error", rid)
-        raise HTTPException(status_code=500, detail="Internal proxy error")
+        raise HTTPException(status_code=500, detail="Internal proxy error")