sunnyzjx commited on
Commit
b8d01e0
·
verified ·
1 Parent(s): e44d607

Update ui_components.py

Browse files
Files changed (1) hide show
  1. ui_components.py +53 -53
ui_components.py CHANGED
@@ -1,53 +1,53 @@
1
- import gradio as gr
2
- from task_manager import get_current_task, select_audio, change_task, tasks
3
- from annotation import save_annotations
4
-
5
-
6
- def create_ui(init_task, username, annotation_results):
7
- """创建 Gradio 界面组件"""
8
- init_inst, init_text, init_audioA_data, init_audioA_rate, init_audioB_data, init_audioB_rate, init_prev_dis, init_next_dis, init_task_num = init_task
9
-
10
- gr.Markdown('<div class="center"><h3>🎵 音频对比标注平台</h3></div>')
11
- user_display = gr.Markdown()
12
-
13
- instruction = gr.Textbox(label="🎯 指令", value=init_inst, interactive=False)
14
- text_box = gr.Textbox(label="📋 转录文本", value=init_text, interactive=False)
15
- task_number = gr.Markdown(value=f"**任务编号: {init_task_num}**", elem_classes="center")
16
-
17
- with gr.Row():
18
- with gr.Column(elem_classes="audio-container"):
19
- audioA = gr.Audio(
20
- label="🔊 音频 A",
21
- value=(init_audioA_rate, init_audioA_data),
22
- interactive=False,
23
- type="numpy"
24
- )
25
- btn_A = gr.Button("✅ 选择音频 A", variant="primary")
26
- with gr.Column(elem_classes="audio-container"):
27
- audioB = gr.Audio(
28
- label="🔊 音频 B",
29
- value=(init_audioB_rate, init_audioB_data),
30
- interactive=False,
31
- type="numpy"
32
- )
33
- btn_B = gr.Button("✅ 选择音频 B", variant="primary")
34
-
35
- with gr.Row():
36
- btn_prev = gr.Button("⬅️ 上一题", interactive=not init_prev_dis)
37
- btn_next = gr.Button("➡️ 下一题", interactive=not init_next_dis)
38
-
39
- return {
40
- "user_display": user_display,
41
- "instruction": instruction,
42
- "text_box": text_box,
43
- "task_number": task_number,
44
- "audioA": audioA,
45
- "audioB": audioB,
46
- "btn_A": btn_A,
47
- "btn_B": btn_B,
48
- "btn_prev": btn_prev,
49
- "btn_next": btn_next,
50
- "select_audio": select_audio,
51
- "change_task": change_task,
52
- "save_annotations": lambda u, a: save_annotations(u, a, tasks)
53
- }
 
1
+ import gradio as gr
2
+ from task_manager import get_current_task, select_audio, change_task, tasks
3
+ from annotation import save_annotations
4
+
5
+
6
+ def create_ui(init_task, username, annotation_results):
7
+ """创建 Gradio 界面组件"""
8
+ init_inst, init_text, init_audioA_data, init_audioA_rate, init_audioB_data, init_audioB_rate, init_prev_dis, init_next_dis, init_task_num = init_task
9
+
10
+ gr.Markdown('<div class="center"><h2>🎵 音频对比标注平台</h2></div>')
11
+ user_display = gr.Markdown()
12
+
13
+ instruction = gr.Textbox(label="🎯 指令", value=init_inst, interactive=False)
14
+ text_box = gr.Textbox(label="📋 转录文本", value=init_text, interactive=False)
15
+ task_number = gr.Markdown(value=f"**任务编号: {init_task_num}**", elem_classes="center")
16
+
17
+ with gr.Row():
18
+ with gr.Column(elem_classes="audio-container"):
19
+ audioA = gr.Audio(
20
+ label="🔊 音频 A",
21
+ value=(init_audioA_rate, init_audioA_data),
22
+ interactive=False,
23
+ type="numpy"
24
+ )
25
+ btn_A = gr.Button("✅ 选择音频 A", variant="primary")
26
+ with gr.Column(elem_classes="audio-container"):
27
+ audioB = gr.Audio(
28
+ label="🔊 音频 B",
29
+ value=(init_audioB_rate, init_audioB_data),
30
+ interactive=False,
31
+ type="numpy"
32
+ )
33
+ btn_B = gr.Button("✅ 选择音频 B", variant="primary")
34
+
35
+ with gr.Row():
36
+ btn_prev = gr.Button("⬅️ 上一题", interactive=not init_prev_dis)
37
+ btn_next = gr.Button("➡️ 下一题", interactive=not init_next_dis)
38
+
39
+ return {
40
+ "user_display": user_display,
41
+ "instruction": instruction,
42
+ "text_box": text_box,
43
+ "task_number": task_number,
44
+ "audioA": audioA,
45
+ "audioB": audioB,
46
+ "btn_A": btn_A,
47
+ "btn_B": btn_B,
48
+ "btn_prev": btn_prev,
49
+ "btn_next": btn_next,
50
+ "select_audio": select_audio,
51
+ "change_task": change_task,
52
+ "save_annotations": lambda u, a: save_annotations(u, a, tasks)
53
+ }