Update app.py

app.py

@@ -1,13 +1,14 @@
 import os
+import json
 import asyncio
 import aiohttp
-import json
+from contextlib import asynccontextmanager
 from fastapi import FastAPI, Request, Response
 from fastapi.responses import StreamingResponse, JSONResponse
 
 TARGET_BASE_URL = "https://api.deepinfra.com/v1/openai"
-REAL_API_KEY = os.environ["REAL_API_KEY"]        # 从 Space secrets 读取
-FAKE_KEY = os.environ.get("FAKE_KEY", "123456")  # 本地访问时的鉴权
+REAL_API_KEY = os.environ["REAL_API_KEY"]
+FAKE_KEY = os.environ.get("FAKE_KEY", "123456")
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -26,7 +27,7 @@ async def proxy_endpoint(path: str = "", request: Request = None):
             "Access-Control-Allow-Headers": "Authorization, Content-Type"
         })
 
-    # 鉴权：只允许携带正确 FAKE_KEY 的请求
+    # 验证本地传来的假 key
     auth_header = request.headers.get("authorization", "")
     if auth_header.lower() != f"bearer {FAKE_KEY}":
         return JSONResponse(status_code=401, content={"error": "Invalid fake key"})
@@ -90,4 +91,4 @@ async def proxy_endpoint(path: str = "", request: Request = None):
                 "content-type": resp.headers.get("content-type", "application/json"),
                 "access-control-allow-origin": "*"
             }
-        )
+        )