Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import config
 from ui_components import create_ui
 from annotation import load_annotations
-from task_manager import get_current_task, get_current_task_with_annotations, get_total_tasks, select_audio, change_task
+from task_manager import get_current_task, get_current_task_with_annotations, get_total_tasks
 
 css = """
 .center { text-align: center; }
@@ -28,12 +28,11 @@ def set_user_info(request: gr.Request):
     """设置用户信息到 State 并加载用户特定的标注"""
     username = request.username if hasattr(request, 'username') else "unknown"
     annotation_results = load_annotations(username)
-    current_task = 0  # 每个用户都从任务0开始
     print(f"加载用户 {username} 的标注结果：{annotation_results}")
 
     # 获取当前任务信息，并考虑已有的标注结果
     inst, text, audioA_update, audioB_update, prev_disabled, next_disabled, task_num = get_current_task_with_annotations(
-        current_task, annotation_results)
+        annotation_results)
 
     # 获取总任务数
     total_tasks = get_total_tasks()
@@ -44,7 +43,6 @@ def set_user_info(request: gr.Request):
     # 返回所有需要初始化的组件值
     return (
         username,
-        current_task,  # 新增：返回用户的 current_task
         annotation_results,
         inst,  # instruction
         text,  # text_box
@@ -60,17 +58,15 @@ if __name__ == "__main__":
     print("启动应用...")
     with gr.Blocks(css=css) as demo:
         username = gr.State(value="unknown")
-        current_task = gr.State(value=0)  # 新增：每个用户的 current_task 状态
         annotation_results = gr.State(value={})
-        ui_components = create_ui(get_current_task(0), username, annotation_results)
+        ui_components = create_ui(get_current_task(), username, annotation_results)
 
-        # 修改 demo.load，添加 current_task 输出
+        # 修改 demo.load，移除 user_display，保留 task_number
         demo.load(
             set_user_info,
             inputs=None,
             outputs=[
                 username,
-                current_task,  # 新增输出
                 annotation_results,
                 ui_components["instruction"],
                 ui_components["text_box"],
@@ -82,33 +78,33 @@ if __name__ == "__main__":
             ]
         )
 
-        # 修改按钮绑定，添加 current_task 参数
+        # 其余按钮绑定保持不变
         ui_components["btn_A"].click(
-            select_audio,
-            inputs=[gr.State("A"), ui_components["audioA"], ui_components["audioB"], current_task, annotation_results, username],
+            ui_components["select_audio"],
+            inputs=[gr.State("A"), ui_components["audioA"], ui_components["audioB"], annotation_results, username],
             outputs=[ui_components["audioA"], ui_components["audioB"], annotation_results]
         )
 
         ui_components["btn_B"].click(
-            select_audio,
-            inputs=[gr.State("B"), ui_components["audioA"], ui_components["audioB"], current_task, annotation_results, username],
+            ui_components["select_audio"],
+            inputs=[gr.State("B"), ui_components["audioA"], ui_components["audioB"], annotation_results, username],
             outputs=[ui_components["audioA"], ui_components["audioB"], annotation_results]
         )
 
         ui_components["btn_prev"].click(
-            change_task,
-            inputs=[gr.State("prev"), current_task, annotation_results, username],
+            ui_components["change_task"],
+            inputs=[gr.State("prev"), annotation_results, username],  # 添加 username 参数
             outputs=[ui_components["instruction"], ui_components["text_box"], ui_components["audioA"],
                      ui_components["audioB"], ui_components["btn_prev"], ui_components["btn_next"],
-                     ui_components["task_number"], current_task, annotation_results]  # 新增 current_task 输出
+                     ui_components["task_number"], annotation_results]  # 保持 task_number
         )
 
         ui_components["btn_next"].click(
-            change_task,
-            inputs=[gr.State("next"), current_task, annotation_results, username],
+            ui_components["change_task"],
+            inputs=[gr.State("next"), annotation_results, username],  # 添加 username 参数
             outputs=[ui_components["instruction"], ui_components["text_box"], ui_components["audioA"],
                      ui_components["audioB"], ui_components["btn_prev"], ui_components["btn_next"],
-                     ui_components["task_number"], current_task, annotation_results]  # 新增 current_task 输出
+                     ui_components["task_number"], annotation_results]  # 保持 task_number
         )
 
-    demo.launch(auth=config.ANNOTATOR, share=True, ssr_mode=False)
+    demo.launch(auth=config.ANNOTATOR, share=True,ssr_mode=False)