Fix OAuth login: improve username detection and fix Gradio 6.0 theme parameter

app.py

@@ -213,23 +213,58 @@ def check_user_access(request: gr.Request) -> Tuple[str, bool]:
         return "", False
     
     # 获取登录用户名
-    # 在 Hugging Face Space 中，用户名可能在不同的属性中
+    # 在 Hugging Face Space 中，OAuth 登录后用户名可能在多个地方
     username = None
-    if hasattr(request, 'username'):
+    
+    # 方法1: 直接从 request.username 获取（Gradio 6.0）
+    if hasattr(request, 'username') and request.username:
         username = request.username
-    elif hasattr(request, 'client') and hasattr(request.client, 'username'):
+        logger.info(f"从 request.username 获取用户名: {username}")
+    
+    # 方法2: 从 request.client.username 获取
+    elif hasattr(request, 'client') and hasattr(request.client, 'username') and request.client.username:
         username = request.client.username
+        logger.info(f"从 request.client.username 获取用户名: {username}")
+    
+    # 方法3: 从 cookies 中获取（HF OAuth 可能设置 cookie）
+    elif hasattr(request, 'cookies'):
+        # 检查常见的 HF cookie 名称
+        for cookie_name in ['hf_user', 'username', 'user', 'hf_username']:
+            if cookie_name in request.cookies:
+                username = request.cookies[cookie_name]
+                logger.info(f"从 cookie {cookie_name} 获取用户名: {username}")
+                break
+    
+    # 方法4: 从 headers 中获取（HF OAuth 常用方式）
     elif hasattr(request, 'headers'):
-        # 尝试从 headers 中获取
-        auth_header = request.headers.get('authorization', '')
-        if auth_header:
-            # 这里可能需要解析 JWT token，简化处理
-            pass
+        # HF OAuth 通常在 headers 中设置 X-User
+        header_value = None
+        for header_name in ['X-User', 'x-user', 'X-HF-User', 'x-hf-user', 'X-Username', 'x-username']:
+            if header_name in request.headers:
+                header_value = request.headers[header_name]
+                logger.info(f"从 header {header_name} 获取: {header_value}")
+                username = header_value
+                break
+        
+        # 如果没有找到，尝试从 authorization header 解析
+        if not username and 'authorization' in request.headers:
+            auth_header = request.headers.get('authorization', '')
+            logger.info(f"找到 authorization header: {auth_header[:50]}...")
+            # 这里可以解析 JWT token 获取用户名（如果需要）
     
-    # 如果 Space 启用了 OAuth，可以通过环境变量获取
+    # 方法5: 从环境变量获取（如果 Space 设置了）
     if not username:
-        # 在 HF Space 中，可以通过环境变量获取
         username = os.getenv('HF_USERNAME', '')
+        if username:
+            logger.info(f"从环境变量 HF_USERNAME 获取用户名: {username}")
+    
+    # 调试：打印 request 对象的所有属性
+    if not username:
+        logger.warning("无法获取用户名，request 对象属性:")
+        if hasattr(request, '__dict__'):
+            logger.warning(f"request.__dict__: {request.__dict__}")
+        logger.warning(f"request 类型: {type(request)}")
+        logger.warning(f"request dir: {[attr for attr in dir(request) if not attr.startswith('_')]}")
     
     if not username:
         return "", False
@@ -449,15 +484,16 @@ def submit_evaluation(
         return f"❌ 提交评分失败: {str(e)}"
 
 
-def get_user_info(request: gr.Request) -> str:
+def get_user_info(request: gr.Request = None) -> str:
     """获取用户信息"""
+    # 如果没有 request，返回提示
     if not request:
-        return "❌ 请登录 Hugging Face 账户"
+        return "⏳ 正在检查登录状态..."
     
     username, has_access = check_user_access(request)
     
     if not username:
-        return "❌ 请登录 Hugging Face 账户\n\n提示：在 Space Settings 中启用 OAuth 登录"
+        return "❌ 请先登录 Hugging Face 账户\n\n提示：点击右上角的 'Login with Hugging Face' 按钮登录"
     
     calls_today = dataset_manager.get_user_calls_today(username) if dataset_manager else 0
     remaining = max(0, MAX_DAILY_CALLS - calls_today)
@@ -467,7 +503,7 @@ def get_user_info(request: gr.Request) -> str:
 
 
 # 创建 Gradio 界面
-with gr.Blocks(title="Video Model Evaluator", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Video Model Evaluator") as demo:
     gr.Markdown("""
     # 🎬 视频生成模型评估系统
     
@@ -615,11 +651,13 @@ with gr.Blocks(title="Video Model Evaluator", theme=gr.themes.Soft()) as demo:
     )
     
     # 页面加载时更新用户信息
-    # 注意：在 Gradio 4.x 中，需要使用正确的 request 参数
+    # 注意：demo.load 可能无法获取 request，所以先显示提示
+    # 用户操作时会自动更新（通过 generate_btn 点击事件）
     demo.load(
-        fn=get_user_info,
+        fn=lambda: "⏳ 请先登录 Hugging Face 账户",
         inputs=[],
-        outputs=[user_info]
+        outputs=[user_info],
+        show_progress=False
     )
     
     # 添加说明
@@ -642,6 +680,7 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False
+        share=False,
+        theme=gr.themes.Soft()  # Gradio 6.0: theme 参数放在 launch() 中
     )