Update app.py

app.py

@@ -99,7 +99,7 @@ def save_current(source, target, rater_selector, error_span, category, subcatego
     system = data.loc[current_index, "system"]
     lp = data.loc[current_index, "lp"]
     doc = data.loc[current_index, "doc"]
-    id = data.loc[current_index, "id"]
+    id = int(data.loc[current_index, "id"])
     reference = data.loc[current_index, "reference"]
     
     
@@ -138,7 +138,7 @@ def save_and_next(source, target, score, rater_selector):
     system = data.loc[current_index, "system"]
     lp = data.loc[current_index, "lp"]
     doc = data.loc[current_index, "doc"]
-    id = data.loc[current_index, "id"]
+    id = int(data.loc[current_index, "id"])
     reference = data.loc[current_index, "reference"]
         
     new_entry = {
@@ -166,18 +166,19 @@ def save_and_next(source, target, score, rater_selector):
 def update_file_selection(selected_file):
     global data_path, data, current_index, annotations_file, score_file
     data_path = os.path.join(current_dir, selected_file)
-    data = pd.read_csv(data_path, dtype={"id": "Int64"})
-    
-    all_ids = get_all_ids()  
-    current_index = int(all_ids[0])  # 設定起始 index 為最小的 id
+    data = pd.read_csv(data_path)
     
+    id_list = [str(id) for id in sorted(data["id"].unique())]  # 轉為字串，確保 Gradio Dropdown 兼容
+    min_id = id_list[0]  # 取得最小的 ID
+
+    current_index = data.index[data["id"] == int(min_id)].tolist()[0]  # 設定 current_index
+
     file_base_name = os.path.splitext(selected_file)[0]
     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() + ("", all_ids, str(current_index), f"已加載檔案：{selected_file}")
+    return get_current_text() + ("", gr.update(choices=id_list, value=min_id), str(current_index), f"已加載檔案：{selected_file}")
 
-    
 
 def update_index_selection(selected_index):
     global current_index, data
@@ -258,6 +259,7 @@ with gr.Blocks() as demo:
             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])