Update app.py

app.py

@@ -45,7 +45,7 @@ def save_to_json(entry: dict, json_file: Path):
             f.write("\n")
         # scheduler.push_to_hub(commit_message=f"更新檔案 {json_file.name}")
 
-def save_current(source, target, rater_selector, error_span, correction_text, category, subcategory, severity, other):
+def save_current(source, target, rater_selector, error_span, category, subcategory, severity, other):
     global annotations_file
     if category != "Non-translation":
         category_value = f"{category}/{subcategory}"
@@ -57,21 +57,21 @@ def save_current(source, target, rater_selector, error_span, correction_text, ca
         "target": target,
         "rater": rater_selector,
         "error_span": error_span,
-        "correction_text": correction_text,
         "category": category_value,
         "severity": severity,
         "other": other if other else "",
     }
     save_to_json(new_entry, annotations_file)
-    return source, target, "", "", f"當前資料已保存到 {annotations_file.name}，請繼續標記！"
+    return source, target, "", f"當前資料已保存到 {annotations_file.name}，請繼續標記！"
 
-def save_and_next(source, target, rater_selector, error_span, correction_text, category, subcategory, severity, other, score):
+def save_and_next(source, target, correction_text, score, rater_selector):
     global current_index, data, score_file
     new_entry = {
-        "source": source,
-        "target": target,
-        "rater": rater_selector,
+        "src": source,
+        "mt": target,
+        "ref": correction_text,
         "score": score,
+        "rater": rater_selector,
     }
     save_to_json(new_entry, score_file)
 
@@ -129,16 +129,19 @@ with gr.Blocks() as demo:
         with gr.Row():
             source = gr.Textbox(label="原始文本", lines=5, interactive=False)
             target = gr.Textbox(label="翻譯文本", lines=5, interactive=False)
-        with gr.Row():
-            error_span = gr.Textbox(label="錯誤區間", lines=2, placeholder="請複製並貼上翻譯中的錯誤文本 (如無錯誤則不需)")
-            correction_text = gr.Textbox(label="正確內容", lines=2, placeholder="輸入修正後的正確內容 (如無錯誤則不需)")
+        # with gr.Row():
+        #     error_span = gr.Textbox(label="錯誤區間", lines=2, placeholder="請複製並貼上翻譯中的錯誤文本 (如無錯誤則不需)")
+        #     correction_text = gr.Textbox(label="正確內容", lines=2, placeholder="輸入修正後的正確內容 (如無錯誤則不需)")
 
         with gr.Row(variant='panel', equal_height=True):
-            severity = gr.Radio(label="錯誤嚴重程度", choices=["No-error", "Minor", "Major"], value="No-error")
-            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="若無法歸類，請填寫其他")
-            save_current_button = gr.Button("保存並繼續標記當前資料")
+            with gr.column(scale=1):
+                error_span = gr.Textbox(label="錯誤區間", lines=2, placeholder="請輸入翻譯中的文本錯誤區間 (如無錯誤則不需)")
+            with gr.column(scale=1):
+                severity = gr.Radio(label="錯誤嚴重程度", choices=["No-error", "Minor", "Major"], value="No-error")
+                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="若無法歸類，請填寫其他")
+                save_current_button = gr.Button("保存並繼續標記當前資料")
 
         def update_subcategories(selected_category):
             subcategories = categories[selected_category]
@@ -150,14 +153,15 @@ with gr.Blocks() as demo:
         category.change(update_subcategories, inputs=[category], outputs=[subcategory])
 
         with gr.Row(variant='panel', equal_height=True):
+            correction_text = gr.Textbox(label="正確內容", lines=2, placeholder="請輸入完整正確的翻譯 (如無錯誤則不需)")
             score = gr.Slider(label="翻譯評分", minimum=0, maximum=100, step=1, value=50)
             save_next_button = gr.Button("保存並顯示下一筆")
 
         status = gr.Textbox(label="當前狀態", lines=1, interactive=False)
 
         file_selector.change(update_file_selection, inputs=[file_selector], outputs=[source, target, error_span, correction_text, status])
-        save_current_button.click(save_current, inputs=[source, target, rater_selector, error_span, correction_text, category, subcategory, severity, other], outputs=[source, target, error_span, correction_text, status])
-        save_next_button.click(save_and_next, inputs=[source, target, rater_selector, error_span, correction_text, category, subcategory, severity, other, score], outputs=[source, target, error_span, correction_text, status])
+        save_current_button.click(save_current, inputs=[source, target, rater_selector, error_span, category, subcategory, severity, other], outputs=[source, target, error_span, status])
+        save_next_button.click(save_and_next, inputs=[source, target, correction_text, score, rater_selector], outputs=[source, target, error_span, correction_text, status])
 
         original, translated = get_current_text()
         source.value = original