Update app.py

app.py

@@ -77,12 +77,9 @@ score_file = DATASET_DIR / f"test_score-{uuid4()}.json"
 
 def get_current_text():
     global current_index, data
-    if current_index < len(data):
-        source = data.loc[current_index, "source"]
-        target = data.loc[current_index, "target"]
-        return source, target
-    else:
-        return "已完成所有文本標記", "已完成所有文本標記"
+    source = data.loc[current_index, "source"]
+    target = data.loc[current_index, "target"]
+    return source, target
 
 def save_to_json(entry: dict, json_file: Path):
     """
@@ -140,6 +137,9 @@ def save_and_next(source, target, score, rater_selector):
     doc = data.loc[current_index, "doc"]
     id = int(data.loc[current_index, "id"])
     reference = data.loc[current_index, "reference"]
+
+    id_list = [str(id) for id in sorted(data["id"].unique())]
+    max_id = int(id_list[-1])  # 取得最大的 ID
         
     new_entry = {
         "system": system,
@@ -155,7 +155,8 @@ def save_and_next(source, target, score, rater_selector):
     save_to_json(new_entry, score_file)
 
     current_index += 1
-    if current_index >= len(data):
+    count = data.index[data["id"] == current_index].tolist()[0]
+    if current_index > max_id:
         return "已完成所有文本標記", "已完成所有文本標記", "", "", f"所有標記已完成並保存到 {score_file.name}！"
         
     next_source, next_target = get_current_text()