Update app.py

app.py

@@ -787,7 +787,7 @@ def prep_and_save_data(index, skip_sample, completed: bool):
         **(st.session_state.data.iloc[index][COLS_TO_SAVE].to_dict() if 0 <= index < len(st.session_state.data) else {}),
         **{k: st.session_state[k + str(index)] for k in st.session_state.data_inputs_keys},
         'skip': skip_sample,
-        'completed': completed,
+        'completed': int(completed),
     }
     
     for f in iter_all_input_fields():
@@ -884,10 +884,26 @@ if user_id_from_url and not st.session_state.get("user_id"):
 # current index
 if 'current_index' not in st.session_state:
     start_index = get_start_index()
-    if start_index < len(st.session_state.data)-1:
-        st.session_state.current_index = start_index
+    target_index = start_index
+
+    if start_index > 0:
+        last_idx = start_index - 1
+        last_path = f"{output_repo_path}/{get_base_path()}/{last_idx}.json"
+        if hf_fs.exists(last_path):
+            try:
+                with hf_fs.open(last_path, "rb") as f:
+                    last_data = json.load(f)
+            except Exception:
+                last_data = {}
+            completed_val = last_data.get("completed", 1)
+            is_completed = bool(completed_val)
+            if not is_completed:
+                target_index = last_idx
+    
+    if target_index < len(st.session_state.data)-1:
+        st.session_state.current_index = target_index
     else:
-        st.session_state.current_index = start_index+1
+        st.session_state.current_index = target_index+1
     st.session_state.form_displayed = -2
 
 if 'step' not in st.session_state: