Update main.py

main.py

@@ -5,6 +5,7 @@ import socket
 
 app = FastAPI()
 
+# Your real TypeGPT API key
 REAL_API_KEY = "sk-kwWVUQPDsLemvilLcyzSSWRzo8sctCzxzlbdN0ZC5ZUCCv0m"
 BASE_URL = "https://fast.typegpt.net"
 PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
@@ -21,13 +22,15 @@ async def startup_event():
 
 @app.api_route("/{path:path}", methods=["GET", "POST"])
 async def proxy(request: Request, path: str, authorization: str = Header(None)):
+    # Validate Authorization header
     if not authorization or not authorization.startswith("Bearer "):
         raise HTTPException(status_code=401, detail="Missing or malformed Authorization header.")
-
+    
     token = authorization.replace("Bearer ", "").strip()
     if token != PUBLIC_AUTH_TOKEN:
         raise HTTPException(status_code=401, detail="Invalid Authorization token. Use 'TypeGPT-Free4ALL'.")
 
+    # Build backend request
     target_url = f"{BASE_URL}/{path}"
     headers = {
         "Authorization": f"Bearer {REAL_API_KEY}",
@@ -44,6 +47,7 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
     print(f"🔁 Stream mode: {is_stream}")
 
     if is_stream:
+        # Stream response immediately, as chunks arrive
         async def stream_generator():
             async with httpx.AsyncClient(timeout=None) as client:
                 async with client.stream(
@@ -52,9 +56,8 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
                     headers=headers,
                     content=body
                 ) as upstream_response:
-                    async for chunk in upstream_response.aiter_lines():
-                        if chunk.strip():  # skip blank lines
-                            yield f"{chunk}\n"
+                    async for chunk in upstream_response.aiter_raw():
+                        yield chunk  # raw bytes streamed directly
 
         return StreamingResponse(
             stream_generator(),
@@ -63,6 +66,7 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
         )
 
     else:
+        # Normal JSON response
         async with httpx.AsyncClient(timeout=60) as client:
             try:
                 response = await client.request(