Update app.py

app.py

@@ -129,6 +129,8 @@ def save_current(source, target, rater_selector, error_span, category, subcatego
         "rater": rater_selector,
     }
     save_to_json(new_entry, annotations_file)
+
+    # [error_span, status]
     return "", f"當前資料已保存到 {annotations_file.name}，請繼續標記！"
 
 def save_and_next(source, target, score, rater_selector):
@@ -155,9 +157,10 @@ def save_and_next(source, target, score, rater_selector):
     current_index += 1
     if current_index < len(data):
         next_source, next_target = get_current_text()
+        # [source, target, error_span, current_index_display, status]
         return next_source, next_target, "", str(current_index), f"分數已保存到 {score_file.name}，請繼續下一筆！"
     else:
-        return "已完成所有文本標記", "已完成所有文本標記", "", str(current_index), 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
@@ -169,7 +172,9 @@ 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"
 
-    return get_current_text() + ("", "", str(current_index), f"已加載檔案：{selected_file}")
+    # [source, target, error_span, index_selector, current_index_display, status]
+    return get_current_text() + ("", get_all_ids(), str(current_index), f"已加載檔案：{selected_file}")
+    
 
 def update_index_selection(selected_index):
     global current_index, data
@@ -177,6 +182,7 @@ def update_index_selection(selected_index):
         selected_index = int(selected_index)
         if selected_index in data["id"].values:
             current_index = data.index[data["id"] == selected_index].tolist()[0]
+            # [source, target, current_index_display, status]
             return get_current_text() + (str(current_index), f"已跳轉至 id: {selected_index}")
     except:
         return "錯誤", "錯誤", str(current_index), "選擇的索引無效，請重新選擇"