Feat: Save progress

app.py

@@ -214,7 +214,8 @@ total_annotations = len(file_list)
 # Initialize an empty DataFrame to store annotations
 annotations = pd.DataFrame(columns=['sample_id', 'sentence', 'emotion', 'confidence', 'comments'])
 current_index = {"index": 0}  # Dictionary to allow modifying inside functions
-
+ann_completed = gr.Number(0, visible=False)
+total = gr.Number(total_annotations, visible=False)
 def get_storage():
     #source: https://discuss.huggingface.co/t/accessing-data-folder-of-persistent-storage/46681/2
     files = [
@@ -232,6 +233,20 @@ def get_storage():
     return files, f"{usage/(1024.0 ** 3):.3f}GB"
 
 
+def load_first_example(index, participant_id):
+    """ Loads and first example and updates index"""
+
+    path_ann = f'{persistent_storage}/{participant_id}_annotations.csv'
+    print(path_ann)
+    if os.path.exists(path_ann):
+        annotations = pd.read_csv(path_ann, index_col = 0)
+        current_index['index'] = len(annotations)
+        print('path was found')
+        print(annotations)
+        ann_completed = len(annotations) - 1
+        print(current_index)
+    return load_example(current_index["index"])
+
 def load_example(index):
     """Loads the example in row #index from dataframe file_list. 
     If there are any annotations it will give those values to the annotation dataframe"""
@@ -316,7 +331,7 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
     with gr.Tab("Instructions", elem_id = 'instructions'):
         instructions = gr.HTML(intro_html, padding = False)
         
-        with gr.Blocks("Audio examples"):
+        with gr.Blocks():
             description = gr.HTML(examples_explanation, padding = False)
 
             with gr.Accordion(label = "Neutral", open= False):
@@ -340,8 +355,7 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
 
     with gr.Tab("Annotation Interface"):
 
-        ann_completed = gr.Number(0, visible=False)
-        total = gr.Number(total_annotations, visible=False)
+
         
         # Row with progress bar
         gr.HTML("""
@@ -397,8 +411,8 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         lets_go.click(None, [], [ann_completed, total], js = js_progress_bar)
         lets_go.click(deactivate_participant_id, [participant_id, lets_go], [participant_id, lets_go])
         lets_go.click(activate_elements, [emotions, confidence, comments, next_button, previous_button], [emotions, confidence, comments, next_button, previous_button])
-        lets_go.click(load_example, inputs = [gr.Number(current_index["index"], visible = False)], outputs = [sentence_text, audio_player, emotions, confidence, ann_completed, comments])
-
+        lets_go.click(load_first_example, inputs = [gr.Number(current_index["index"], visible = False), participant_id], outputs = [sentence_text, audio_player, emotions, confidence, ann_completed, comments])
+        
     with gr.Tab("Access Files"):
         with gr.Row():
             with gr.Column():