Update app.py

app.py

@@ -202,8 +202,7 @@ def get_start_index():
         return -2
 
 def read_data():
-    assert st.session_state.batch, "Batch not provided"
-    with hf_fs.open(input_repo_path + '/' + to_annotate_file_name.format(batch=st.session_state.batch)) as f:
+    with hf_fs.open(input_repo_path + '/' + to_annotate_file_name) as f:
         return pd.read_csv(f)
 
 def read_saved_data():
@@ -224,7 +223,7 @@ def save_data(data):
         f.write(json.dumps(data))
 
 def get_base_path():
-    return f"{st.session_state.batch}/{st.session_state.user_id}"
+    return f"{st.session_state.user_id}"
 
 def get_path():
     return f"{get_base_path()}/{st.session_state.current_index}.json"
@@ -426,7 +425,6 @@ def prep_and_save_data(index, skip_sample):
     save_data({
         'user_id': st.session_state.user_id,
         'index': st.session_state.current_index,
-        'batch': st.session_state.batch,
         **(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
@@ -473,7 +471,7 @@ if 'unacceptable_response' in st.session_state and st.session_state.unacceptable
     st.stop()
 
 # batch-specific data subset
-st.session_state.batch = get_param_from_url("batch")
+#st.session_state.batch = get_param_from_url("batch")
 
 # Load the data to annotate
 if 'data' not in st.session_state: