Update app.py

app.py

@@ -229,11 +229,18 @@ else:
     hf_fs = LocalFileSystem()
 
 def get_start_index():
-    if hf_fs.exists(output_repo_path + '/' + get_base_path()):
-        files = hf_fs.ls(output_repo_path + '/' + get_base_path())
-        return len(files) #-1
-    else:
+    base_dir = f"{output_repo_path}/{get_base_path()}"
+    try:
+        files = hf_fs.ls(base_dir)
+    except Exception as e:
         return -2
+    return len(files)
+    
+    #if hf_fs.exists(output_repo_path + '/' + get_base_path()):
+    #    files = hf_fs.ls(output_repo_path + '/' + get_base_path())
+    #    return len(files) #-1
+    #else:
+    #    return -2
 
 def read_data():
     with hf_fs.open(input_repo_path + '/' + to_annotate_file_name) as f:
@@ -900,16 +907,16 @@ if 'current_index' not in st.session_state:
     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
+        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