update

app.py

@@ -302,8 +302,23 @@ with demo:
 
         # 提交按钮 (登录功能暂时注释)
         with gr.Row():
-            # login_button = gr.LoginButton("🔐 Login with HuggingFace", size="lg")  # 暂时注释登录功能
+            login_button = gr.LoginButton("🔐 Login with HuggingFace", size="lg")
             submit_button = gr.Button("Submit Results", variant="primary", size="lg")
+        # 登录状态与用户信息
+        profile_state = gr.State()
+        login_status = gr.Markdown(visible=True)
+
+        def on_login(profile: gr.OAuthProfile):
+            try:
+                if profile and getattr(profile, "username", None):
+                    name = profile.username
+                    text = f"✅ 已登录: **{name}**"
+                else:
+                    text = "❌ 登录失败，请重试"
+                return profile, text
+            except Exception:
+                return None, "❌ 登录失败，请重试"
+        login_button.click(on_login, inputs=[], outputs=[profile_state, login_status])
         
         # 进度显示和结果显示区域
         progress_info = gr.HTML()  # 进度显示区域
@@ -331,12 +346,17 @@ with demo:
             """
             return progress_html
         
-        def handle_submission(file_upload, org_name, email, progress=gr.Progress()):
+        def handle_submission(file_upload, org_name, email, user_profile, progress=gr.Progress()):
             try:
                 # 步骤1: 基本验证
                 progress(0.1, desc="验证提交信息...")
                 yield show_progress(1, "验证提交信息"), ""
                 
+                # 校验登录
+                if user_profile is None or getattr(user_profile, "username", None) is None:
+                    yield "", format_error("请先登录 HuggingFace 账号后再提交")
+                    return
+
                 if not file_upload:
                     yield "", format_error("请选择要上传的文件")
                     return
@@ -360,6 +380,12 @@ with demo:
                 import time
                 time.sleep(0.5)  # 让用户看到进度更新
                 
+                # 用户资格检查（账号年龄/频率/重复提交）
+                eligible, msg = check_user_submission_eligibility(user_profile, org_name)
+                if not eligible:
+                    yield "", format_error(msg)
+                    return
+
                 # 步骤3: 上传到OSS
                 progress(0.6, desc="上传文件到OSS...")
                 yield show_progress(3, "上传文件到OSS存储"), ""
@@ -374,6 +400,12 @@ with demo:
                 
                 time.sleep(0.5)  # 让用户看到完成状态
                 
+                # 记录提交历史
+                try:
+                    record_user_submission(user_profile, org_name, email)
+                except Exception:
+                    pass
+
                 # 生成成功信息
                 success_info = f"""
 <div style="background-color: #d4edda; border: 1px solid #c3e6cb; border-radius: 5px; padding: 15px; margin: 10px 0;">
@@ -397,7 +429,7 @@ with demo:
         
         submit_button.click(
             handle_submission,
-            inputs=[file_upload, org_textbox, email_textbox],
+            inputs=[file_upload, org_textbox, email_textbox, profile_state],
             outputs=[progress_info, submission_result],
             show_progress=True
         )