Update main.py

main.py

@@ -3,31 +3,30 @@ import httpx
 
 app = FastAPI()
 
-# Internal secret key (never exposed to users)
 REAL_API_KEY = "sk-qO9N6kQEEULMWtF4YGVlTTSjIPllEm1h1wfEBzSmnSbxiXwe"
 BASE_URL = "https://fast.typegpt.net"
-
-# Public API key that users must send in the Authorization header
 PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
 
 @app.api_route("/{path:path}", methods=["GET", "POST"])
 async def proxy(request: Request, path: str, authorization: str = Header(None)):
-    # Validate public access token
-    if authorization != PUBLIC_AUTH_TOKEN:
+    # Remove 'Bearer ' prefix if present
+    if not authorization or not authorization.startswith("Bearer "):
+        raise HTTPException(status_code=401, detail="Missing or malformed Authorization header.")
+
+    token = authorization.replace("Bearer ", "").strip()
+
+    # Check if it's the correct public token
+    if token != PUBLIC_AUTH_TOKEN:
         raise HTTPException(status_code=401, detail="Invalid Authorization token. Use 'TypeGPT-Free4ALL'.")
 
-    # Construct the target URL
     target_url = f"{BASE_URL}/{path}"
 
-    # Copy headers from request and replace Authorization with real key
     headers = dict(request.headers)
     headers["Authorization"] = f"Bearer {REAL_API_KEY}"
-    headers.pop("host", None)  # remove 'host' header if present
+    headers.pop("host", None)
 
-    # Get the request body
     body = await request.body()
 
-    # Send request to backend (TypeGPT)
     async with httpx.AsyncClient() as client:
         response = await client.request(
             method=request.method,
@@ -36,5 +35,4 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
             headers=headers,
         )
 
-    # Return JSON response from TypeGPT
     return response.json()