Update main.py

main.py

@@ -5,23 +5,23 @@ import socket
 
 app = FastAPI()
 
-# Your internal real API key (DO NOT expose this to users)
+# Real secret API key (used internally, never exposed)
 REAL_API_KEY = "sk-qO9N6kQEEULMWtF4YGVlTTSjIPllEm1h1wfEBzSmnSbxiXwe"
 BASE_URL = "https://fast.typegpt.net"
 
-# Public token that users must use
+# Public key that users must use in Authorization header
 PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
 
-# Print server IP address on startup
+# Show server IP at startup
 @app.on_event("startup")
 async def startup_event():
     try:
         hostname = socket.gethostname()
         server_ip = socket.gethostbyname(hostname)
-        print(f"===== Server Started =====")
+        print("===== Server Started =====")
         print(f"📡 Server IP: {server_ip}")
     except Exception as e:
-        print(f"⚠️ Could not get server IP: {e}")
+        print(f"⚠️ Could not determine server IP: {e}")
 
 @app.api_route("/{path:path}", methods=["GET", "POST"])
 async def proxy(request: Request, path: str, authorization: str = Header(None)):
@@ -30,40 +30,43 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
         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'.")
 
-    # Prepare the backend target URL
+    # Rebuild the target URL
     target_url = f"{BASE_URL}/{path}"
 
-    # Clone headers, replace Authorization
-    headers = dict(request.headers)
-    headers["Authorization"] = f"Bearer {REAL_API_KEY}"
-    headers.pop("host", None)
+    # Use clean headers to avoid Cloudflare issues
+    headers = {
+        "Authorization": f"Bearer {REAL_API_KEY}",
+        "Content-Type": request.headers.get("content-type", "application/json"),
+        "Accept": "application/json",
+        "User-Agent": "FastAPI-Proxy"
+    }
 
-    # Read request body
+    # Read the body
     body = await request.body()
 
-    # Forward request to backend
+    # Log for debugging
+    print(f"🔄 Forwarding to: {target_url}")
+    print(f"📦 Request Body (first 200 chars): {body[:200]}")
+
     async with httpx.AsyncClient() as client:
         try:
             response = await client.request(
                 method=request.method,
                 url=target_url,
-                content=body,
                 headers=headers,
+                content=body,
                 timeout=60
             )
         except httpx.RequestError as e:
-            raise HTTPException(status_code=502, detail=f"Request to backend failed: {e}")
+            raise HTTPException(status_code=502, detail=f"Failed to reach backend: {e}")
 
-    # Debug log
-    print(f"🔄 Forwarded to: {target_url}")
     print(f"↩️ TypeGPT Status: {response.status_code}")
-    print(f"🧾 TypeGPT Response (first 200 chars): {response.text[:200]}")
+    print(f"🧾 Response Snippet: {response.text[:200]}")
 
-    # Try to return JSON or fallback
+    # Return JSON if possible, else fallback
     try:
         return JSONResponse(content=response.json(), status_code=response.status_code)
     except Exception: