Update main.py

main.py

@@ -1,38 +1,51 @@
 from fastapi import FastAPI, Request, Header, HTTPException
 from fastapi.responses import JSONResponse, Response
 import httpx
+import socket
 
 app = FastAPI()
 
-# Your internal real API key for TypeGPT (not exposed to users)
-REAL_API_KEY = "sk-ZMNry2FmYh2JTLibwpBcaF6mw8fR2cDXPxQkNO6qfJkTTIi5"
-BASE_URL = "https://api.typegpt.net"
+# Your internal real API key (DO NOT expose this to users)
+REAL_API_KEY = "sk-qO9N6kQEEULMWtF4YGVlTTSjIPllEm1h1wfEBzSmnSbxiXwe"
+BASE_URL = "https://fast.typegpt.net"
 
-# Public token users must send in their Authorization header
+# Public token that users must use
 PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
 
+# Print server IP address on startup
+@app.on_event("startup")
+async def startup_event():
+    try:
+        hostname = socket.gethostname()
+        server_ip = socket.gethostbyname(hostname)
+        print(f"===== Server Started =====")
+        print(f"📡 Server IP: {server_ip}")
+    except Exception as e:
+        print(f"⚠️ Could not get server IP: {e}")
+
 @app.api_route("/{path:path}", methods=["GET", "POST"])
 async def proxy(request: Request, path: str, authorization: str = Header(None)):
-    # Validate the Authorization header
+    # Validate Authorization token
     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'.")
 
-    # Construct URL to forward to the real backend
+    # Prepare the backend target URL
     target_url = f"{BASE_URL}/{path}"
 
-    # Prepare headers with real API key
+    # Clone headers, replace Authorization
     headers = dict(request.headers)
     headers["Authorization"] = f"Bearer {REAL_API_KEY}"
     headers.pop("host", None)
 
-    # Forward the request body
+    # Read request body
     body = await request.body()
 
+    # Forward request to backend
     async with httpx.AsyncClient() as client:
         try:
             response = await client.request(
@@ -40,17 +53,17 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
                 url=target_url,
                 content=body,
                 headers=headers,
-                timeout=60  # prevent hanging
+                timeout=60
             )
         except httpx.RequestError as e:
             raise HTTPException(status_code=502, detail=f"Request to backend failed: {e}")
 
-    # Log response for debugging
-    print("TypeGPT Response Status:", response.status_code)
-    print("TypeGPT Response Headers:", response.headers)
-    print("TypeGPT Response Content:", response.text[:200])  # limit output
+    # 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]}")
 
-    # Try to return JSON response, fallback to raw
+    # Try to return JSON or fallback
     try:
         return JSONResponse(content=response.json(), status_code=response.status_code)
     except Exception: