Update app.py

app.py

@@ -65,7 +65,7 @@ if not os.path.exists(data_path):
 
 
 # Loading Data-----------------------------------------------------------------------------------------------------------
-data = pd.read_csv(data_path)
+data = pd.read_csv(data_path, dtype={"id": "Int64"}) # 確保 id 為標準 Python int
 
 current_index = 0
 def get_all_ids():
@@ -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 = int(data.loc[current_index, "id"])  # 確保 id 為標準 Python int
+    id = data.loc[current_index, "id"]
     reference = data.loc[current_index, "reference"]
     
     
@@ -136,7 +136,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 = int(data.loc[current_index, "id"])  # 確保 id 為標準 Python int
+    id = data.loc[current_index, "id"]
     reference = data.loc[current_index, "reference"]
         
     new_entry = {
@@ -163,7 +163,7 @@ 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)
-    current_index = 0
+    current_index = int(data["id"].min()) # 每個檔案的最小id不一定由0開始
 
     file_base_name = os.path.splitext(selected_file)[0]
     annotations_file = DATASET_DIR / f"{file_base_name}_annotations-{uuid4()}.json"
@@ -180,16 +180,6 @@ def update_index_selection(selected_index):
             return get_current_text() + (str(current_index), f"已跳轉至 id: {selected_index}")
     except:
         return "錯誤", "錯誤", str(current_index), "選擇的索引無效，請重新選擇"
-def update_index_selection(selected_index):
-    global current_index, data
-    try:
-        selected_index = int(selected_index)
-        if selected_index in data["id"].values:
-            current_index = data.index[data["id"] == selected_index].tolist()[0]
-            return get_current_text() + (str(current_index), f"已跳轉至 id: {selected_index}")
-    except:
-        return "錯誤", "錯誤", str(current_index), "選擇的索引無效，請重新選擇"
-
 
         
 categories = {
@@ -228,9 +218,9 @@ with gr.Blocks() as demo:
                 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)
+                source = gr.Textbox(label="原始文本", lines=15, interactive=False)
             with gr.Column(scale=6):
-                target = gr.Textbox(label="翻譯文本", lines=10, interactive=False)
+                target = gr.Textbox(label="翻譯文本", lines=15, interactive=False)
 
         with gr.Row(variant='panel', equal_height=True):
             with gr.Column(scale=1):