Upload 94 files

app.py

@@ -1,5 +1,5 @@
 # ⚙️ 后端逻辑/核心服务端.py (Hugging Face Spaces app.py)
-from runtime_log_privacy import install_uvicorn_privacy_filters, route_label
+from runtime_log_privacy import install_uvicorn_privacy_filters, route_label, RequestTraceMiddleware
 install_uvicorn_privacy_filters()
 
 from fastapi import FastAPI, File, UploadFile, Form, Depends, HTTPException, Request
@@ -171,7 +171,7 @@ async def validation_exception_handler(request, exc):
 @app.exception_handler(Exception)
 async def global_exception_handler(request, exc):
     """🚀 P3优化：全局异常处理，捕获所有未预期异常"""
-    error_id = f"ERR_{int(time.time())}_{id(exc) % 10000}"
+    error_id = getattr(request.state, 'ranking_request_id', None) or uuid.uuid4().hex
     logger.error(f"Unhandled Exception [{error_id}] | {route_label(request)} | error_type={type(exc).__name__}")
     return JSONResponse(
         status_code=500,
@@ -180,7 +180,8 @@ async def global_exception_handler(request, exc):
             "服务器内部错误，请稍后重试",
             "internal_error",
             error_id=error_id
-        )
+        ),
+        headers={"X-Ranking-Request-Id": error_id}
     )
 
 
@@ -364,8 +365,9 @@ app.add_middleware(
     allow_credentials=False, 
     allow_methods=["*"],
     allow_headers=["*"],
-    expose_headers=["ETag", "X-Ranking-Error-Scope"],
+    expose_headers=["ETag", "X-Ranking-Error-Scope", "X-Ranking-Request-Id"],
 )
+app.add_middleware(RequestTraceMiddleware)
 
 # ==========================================
 # 路由挂载

runtime_log_privacy.py

@@ -1,5 +1,45 @@
 """运行服务器日志不记录请求目标、客户端标识或异常正文。"""
 import logging
+import time
+import uuid
+from types import SimpleNamespace
+
+
+class RequestTraceMiddleware:
+    def __init__(self, app, logger=None):
+        self.app = app
+        self.logger = logger or logging.getLogger('ComfyUI-Ranking.Request')
+
+    async def __call__(self, scope, receive, send):
+        if scope['type'] != 'http':
+            return await self.app(scope, receive, send)
+        request_id = uuid.uuid4().hex
+        scope.setdefault('state', {})['ranking_request_id'] = request_id
+        started = time.monotonic()
+        status = 500
+        completed = False
+
+        async def traced_send(message):
+            nonlocal status, completed
+            if message['type'] == 'http.response.start':
+                status = message['status']
+                message = dict(message)
+                message['headers'] = [(key, value) for key, value in message.get('headers', [])
+                                      if key.lower() != b'x-ranking-request-id']
+                message['headers'].append((b'x-ranking-request-id', request_id.encode('ascii')))
+            await send(message)
+            if message['type'] == 'http.response.body' and not message.get('more_body', False):
+                completed = True
+
+        try:
+            return await self.app(scope, receive, traced_send)
+        finally:
+            method = scope.get('method')
+            if method not in {'GET', 'HEAD', 'POST', 'PUT', 'PATCH', 'DELETE', 'OPTIONS', 'CONNECT', 'TRACE'}:
+                method = 'unknown'
+            self.logger.info('REQUEST_FINISHED | request_id=%s | method=%s | route=%s | status=%d | duration_ms=%.3f | completed=%s',
+                             request_id, method, route_label(SimpleNamespace(scope=scope)), status,
+                             max(0, (time.monotonic() - started) * 1000), completed)
 
 
 def route_label(request):