Update app.py

app.py

@@ -153,9 +153,9 @@ def save_and_next(source, target, score, rater_selector):
     save_to_json(new_entry, score_file)
 
     current_index += 1
-    if current_index + 1 < len(data):
+    if current_index < len(data):
         next_source, next_target = get_current_text()
-        return next_source, next_target, "", "", str(current_index), f"分數已保存到 {score_file.name}，請繼續下一筆！"
+        return next_source, next_target, "", str(current_index), f"分數已保存到 {score_file.name}，請繼續下一筆！"
     else:
         return "已完成所有文本標記", "已完成所有文本標記", "", str(current_index), f"所有標記已完成並保存到 {score_file.name}！"
 
@@ -169,7 +169,7 @@ 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() + ("", "", f"已加載檔案：{selected_file}")
+    return get_current_text() + ("", "0", str(current_index), f"已加載檔案：{selected_file}")
 
 def update_index_selection(selected_index):
     global current_index, data
@@ -215,7 +215,7 @@ with gr.Blocks() as demo:
             with gr.Column(scale=1):
                 rater_selector = gr.Dropdown(label="標註人員", choices=rater, value="rater1")
                 file_selector = gr.Dropdown(label="選擇檔案", choices=csv_files, value="test.csv")
-                index_selector = gr.Dropdown(label="選擇索引", choices=get_all_ids(), value=0)
+                index_selector = gr.Dropdown(label="選擇索引", choices=get_all_ids(), value="0")
                 current_index_display = gr.Textbox(label="當前索引", value=str(current_index), interactive=False)
             with gr.Column(scale=6):
                 source = gr.Textbox(label="原始文本", lines=10, interactive=False)
@@ -248,7 +248,8 @@ with gr.Blocks() as demo:
                 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, status])
+        
+        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])