Fix: Save annotations in persistent storage

app.py

@@ -214,8 +214,10 @@ 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 = [
@@ -233,19 +235,23 @@ def get_storage():
     return files, f"{usage/(1024.0 ** 3):.3f}GB"
 
 
-def load_first_example(index, participant_id):
+def load_first_example(index, participant_id, ann_completed):
     """ Loads and first example and updates index"""
+    global annotations
+    global current_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)
+        annotations = pd.read_csv(path_ann)
+
         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"])
+        print(len(annotations))
+        ann_completed = gr.Number(len(annotations) - 1, visible=False)
+        print(len(annotations))
+    return ann_completed, *load_example(current_index["index"])
 
 def load_example(index):
     """Loads the example in row #index from dataframe file_list. 
@@ -260,10 +266,11 @@ def load_example(index):
         annotations.iloc[index].to_dict() if index < len(annotations) else {"sample_id": row["SAMPLE ID"], "emotion": 'Blank', "confidence": 0,
                                                                             "comments": ''}
     )
+    print(previous_annotation)
     return (sentence, audio_path, previous_annotation['emotion'], previous_annotation['confidence'], current_index['index'] + 1, previous_annotation["comments"])
 
 
-def save_annotation(emotions, confidence, comments, participant_id):
+def save_annotation(emotions, confidence, comments, participant_id, ann_completed):
     """Save the annotation for the current example."""
 
     idx = current_index["index"]
@@ -277,29 +284,30 @@ def save_annotation(emotions, confidence, comments, participant_id):
             [emotions, confidence, comments]
     else:
         annotations.loc[len(annotations)] = [sample_id, sentence, emotions, confidence, comments]
-        ann_completed.value += 1
+        ann_completed = gr.Number(ann_completed + 1, visible=False)
     annotations.to_csv(f"{persistent_storage}/{participant_id}_annotations.csv", index=False)  # Save to a CSV file
     
+    return ann_completed
 
-def next_example(emotions, confidence, comments, participant_id):
+def next_example(emotions, confidence, comments, participant_id, ann_completed):
     """Move to the next example."""
     if emotions == "Blank":
         gr.Warning("Please fill out the emotion section. 'Blank' is not a valid emotion.")
     else:
         
-        save_annotation(emotions, confidence, comments, participant_id)
+        ann_completed = save_annotation(emotions, confidence, comments, participant_id, ann_completed)
         if current_index["index"] < len(file_list) - 1:
             current_index["index"] += 1
-    return load_example(current_index["index"])
+    return ann_completed, *load_example(current_index["index"])
 
-def previous_example(emotion, confidence, comments, participant_id):
+def previous_example(emotion, confidence, comments, participant_id,  ann_completed):
     """Move to the previous example."""
     if emotion.value != "Blank":
-        save_annotation(emotion, confidence, comments, participant_id)
+        ann_completed = save_annotation(emotion, confidence, comments, participant_id,  ann_completed)
         if current_index["index"] > 0:
             current_index["index"] -= 1
-            return load_example(current_index["index"])
-        return load_example(current_index["index"])
+            return ann_completed, *load_example(current_index["index"])
+        return ann_completed, *load_example(current_index["index"])
 
 def deactivate_participant_id(participant_id, lets_go):
     participant_id = gr.Textbox(label='What is your participant ID?', value = participant_id, interactive = False)
@@ -350,14 +358,13 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
                 angry_int_audio = gr.Audio(value=f'{persistent_storage}/emotion_examples/angry_intense.wav', label = "Anger (High Intensity)")
 
             agreement = gr.Checkbox(value = False, label = "I agree", info = "I agree to have my annotations, comments, and questionnaire answers used for research purposes. I understand that any personal information will be anonymized.", interactive = True)
-            
-
 
+    
     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("""
         <div id="myProgress">
         <div id="myBar">
@@ -365,7 +372,7 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         </div>
         </div>
         """, padding = False)
-
+    
         # Row with audio player
         with gr.Row():
             audio_player = gr.Audio(value= 'test.mp3', label="Audio", type="filepath", interactive=False, show_download_button = False, show_share_button = False)
@@ -394,34 +401,33 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         # Go back
         previous_button.click(
             previous_example,
-            inputs=[emotions, confidence, comments, participant_id],
-            outputs=[sentence_text, audio_player, emotions, confidence, ann_completed, comments],
+            inputs=[emotions, confidence, comments, participant_id,  ann_completed],
+            outputs=[ ann_completed, sentence_text, audio_player, emotions, confidence, ann_completed, comments],
         )
 
         # Go to the next example
         next_button.click(
             next_example,
-            inputs=[emotions, confidence, comments, participant_id],
-            outputs=[sentence_text, audio_player, emotions, confidence, ann_completed, comments],
-        )
-        #Update progress bar
-        next_button.click(None, [], [ann_completed, total], js = js_progress_bar)
-        
+            inputs=[emotions, confidence, comments, participant_id, ann_completed],
+            outputs=[ann_completed, sentence_text, audio_player, emotions, confidence, ann_completed, comments],
+        ).then(None, [], [ann_completed, total], js = js_progress_bar)
+
+
         
-        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_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():
-                btn = gr.Button("Run")
-            with gr.Column():
-                files = gr.Files(label="Files")
-                storage = gr.Text(label="Total Usage")
-        btn.click(get_storage, inputs=None, outputs=[files, storage], postprocess=False)
-
+        lets_go.click(load_first_example, inputs = [gr.Number(current_index["index"], visible = False), participant_id, ann_completed], outputs = [ann_completed, sentence_text, audio_player, emotions, confidence, ann_completed, comments]).then( None, [], [ann_completed, total], js = js_progress_bar)
+        #lets_go.click(None, [], [ann_completed, total], js = js_progress_bar)
+    
+    #with gr.Tab("Access Files"):
+    #    with gr.Row():
+    #        with gr.Column():
+    #            btn = gr.Button("Run")
+    #        with gr.Column():
+    #            files = gr.Files(label="Files")
+    #            storage = gr.Text(label="Total Usage")
+    #    btn.click(get_storage, inputs=None, outputs=[files, storage], postprocess=False)
+    
         
 demo.launch()