Update main.py

main.py

@@ -3,39 +3,38 @@ import httpx
 
 app = FastAPI()
 
-# Replace with your actual backend API key and base URL
-API_KEY = "sk-qO9N6kQEEULMWtF4YGVlTTSjIPllEm1h1wfEBzSmnSbxiXwe"
+# Internal secret key (never exposed to users)
+REAL_API_KEY = "sk-qO9N6kQEEULMWtF4YGVlTTSjIPllEm1h1wfEBzSmnSbxiXwe"
 BASE_URL = "https://fast.typegpt.net"
 
-# Public API key for users to use
-PUBLIC_API_KEY = "TypeGPT-Free4ALL"
+# 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, x_api_key: str = Header(None)):
-    # Check if the public API key matches the expected value
-    if x_api_key != PUBLIC_API_KEY:
-        raise HTTPException(status_code=401, detail="Invalid API key. Please use 'TypeGPT-Free4ALL'.")
+async def proxy(request: Request, path: str, authorization: str = Header(None)):
+    # Validate public access token
+    if authorization != PUBLIC_AUTH_TOKEN:
+        raise HTTPException(status_code=401, detail="Invalid Authorization token. Use 'TypeGPT-Free4ALL'.")
 
-    # Reconstruct full URL to target endpoint
+    # Construct the target URL
     target_url = f"{BASE_URL}/{path}"
 
-    # Prepare headers from the incoming request
+    # Copy headers from request and replace Authorization with real key
     headers = dict(request.headers)
-    headers["Authorization"] = f"Bearer {API_KEY}"  # Add your secret API key here
-    headers.pop("host", None)  # Remove host header to avoid conflicts
+    headers["Authorization"] = f"Bearer {REAL_API_KEY}"
+    headers.pop("host", None)  # remove 'host' header if present
 
-    # Get the body of the request if present
+    # Get the request body
     body = await request.body()
 
-    # Make the actual request to the backend API
+    # Send request to backend (TypeGPT)
     async with httpx.AsyncClient() as client:
         response = await client.request(
-            request.method,
-            target_url,
+            method=request.method,
+            url=target_url,
             content=body,
-            headers=headers
+            headers=headers,
         )
 
-    # Return the response back to the user
+    # Return JSON response from TypeGPT
     return response.json()
-