Update app.py

app.py

@@ -51,12 +51,30 @@ def init_keys():
         keys = []
         key_cycle = cycle(keys)
 
+# 获取真实IP地址
+def get_client_ip(request: Request) -> str:
+    # 尝试从各种头部获取IP
+    forwarded_for = request.headers.get("x-forwarded-for")
+    if forwarded_for:
+        return forwarded_for.split(",")[0].strip()
+    
+    real_ip = request.headers.get("x-real-ip")
+    if real_ip:
+        return real_ip
+    
+    return request.client.host
+
 # IP白名单中间件
 @app.middleware("http")
 async def ip_whitelist(request: Request, call_next):
     if Config.WHITELIST_IPS and Config.WHITELIST_IPS[0]:
-        client_ip = request.client.host
+        client_ip = get_client_ip(request)
+        print(f"Request from IP: {client_ip}")
+        print(f"Allowed IPs: {Config.WHITELIST_IPS}")
+        print(f"Request headers: {dict(request.headers)}")
+        
         if client_ip not in Config.WHITELIST_IPS:
+            print(f"Access denied for IP: {client_ip}")
             raise HTTPException(status_code=403, detail="IP not allowed")
     return await call_next(request)
 
@@ -133,6 +151,7 @@ async def chat_completions(request: Request):
         url = f"{Config.OPENAI_API_BASE}/chat/completions"
         
         print(f"Chat request to: {url}")
+        print(f"Request from IP: {get_client_ip(request)}")
         print(f"Headers: {headers}")
         print(f"Body: {json.dumps(body_json)}")
         
@@ -196,6 +215,9 @@ async def proxy(path: str, request: Request):
     # 构建目标URL
     url = f"{Config.OPENAI_API_BASE}/{path}"
     
+    print(f"Proxy request to: {url}")
+    print(f"Request from IP: {get_client_ip(request)}")
+    
     async with httpx.AsyncClient() as client:
         try:
             response = await client.request(