Update app.py

app.py

@@ -160,7 +160,7 @@ def save_and_next(source, target, score, rater_selector):
         # [source, target, error_span, current_index_display, status]
         return next_source, next_target, "", str(current_index), f"分數已保存到 {score_file.name}，請繼續下一筆！"
     else:
-        return "已完成所有文本標記", "已完成所有文本標記", "", "", f"所有標記已完成並保存到 {score_file.name}！"
+        return "已完成所有文本標註", "已完成所有文本標註", "", "", f"所有標記已完成並保存到 {score_file.name}！"
 
 def update_file_selection(selected_file):
     global data_path, data, current_index, annotations_file, score_file
@@ -172,9 +172,8 @@ def update_file_selection(selected_file):
     annotations_file = DATASET_DIR / f"{file_base_name}_annotations-{uuid4()}.json"
     score_file = DATASET_DIR / f"{file_base_name}_score-{uuid4()}.json"
 
-    # [source, target, error_span, index_selector, current_index_display, status]
+    # 刷新index的選項
     return get_current_text() + ("", get_all_ids(), str(current_index), f"已加載檔案：{selected_file}")
-    
 
 def update_index_selection(selected_index):
     global current_index, data
@@ -236,33 +235,19 @@ with gr.Blocks() as demo:
                     category = gr.Dropdown(label="錯誤類別", choices=list(categories.keys()), value="No-error")
                     subcategory = gr.Dropdown(label="子類別", choices=categories["No-error"], value=None)
                     other = gr.Textbox(label="其他子類別", placeholder="若無法歸類，請填寫其他")
-                with gr.Row(equal_height=True):       
-                    severity = gr.Radio(label="錯誤嚴重程度", choices=["No-error", "Minor", "Major"], value="No-error")
-                    save_current_button = gr.Button("保存並繼續標記當前資料")
-
-        with gr.Row(variant='panel', equal_height=True):
-            with gr.Column(scale=8):
-                score = gr.Slider(label="翻譯評分", minimum=0, maximum=100, step=1, value=50)
-            with gr.Column(scale=1):
-                save_next_button = gr.Button("保存並顯示下一筆")
-
-        status = gr.Textbox(label="當前狀態", lines=1, interactive=False)
-
-        def update_subcategories(selected_category):
-            subcategories = categories[selected_category]
-            if subcategories:
-                return gr.update(choices=subcategories, value=subcategories[0])
-            else:
-                return gr.update(choices=[], value=None)
-        
-        file_selector.change(update_file_selection, inputs=[file_selector], outputs=[source, target, error_span, index_selector, current_index_display, status])
-        index_selector.change(update_index_selection, inputs=[index_selector], outputs=[source, target, current_index_display, status])
-        category.change(update_subcategories, inputs=[category], outputs=[subcategory])
-        save_current_button.click(save_current, inputs=[source, target, rater_selector, error_span, category, subcategory, severity, other], outputs=[error_span, status])
-        save_next_button.click(save_and_next, inputs=[source, target, score, rater_selector], outputs=[source, target, error_span, current_index_display, status])
+                with gr.Row(equal_height=True): 
+                    severity = gr.Radio(label="錯誤嚴重程度", choices=["Minor", "Major"], value="Minor")
+                    score = gr.Slider(minimum=0, maximum=100, value=100, step=1, label="分數")
+                    
+        with gr.Row(equal_height=True):
+            with gr.Column(scale=3):
+                save_btn = gr.Button("保存並顯示下一筆")
+            with gr.Column(scale=6):
+                save_status = gr.Textbox(label="狀態", interactive=False, lines=1)
 
-        original, translated = get_current_text()
-        source.value = original
-        target.value = translated
+        # Custom functions
+        file_selector.change(update_file_selection, inputs=[file_selector], outputs=[source, target, error_span, index_selector, current_index_display])
+        index_selector.change(update_index_selection, inputs=[index_selector], outputs=[source, target, current_index_display, save_status])
+        save_btn.click(save_and_next, inputs=[source, target, score, rater_selector], outputs=[source, target, error_span, current_index_display, save_status])
         
 demo.launch()