Feat: Add click counter

app.py

@@ -217,7 +217,7 @@ global file_list
 global total_annotations 
 
 # Initialize an empty DataFrame to store annotations
-annotations = pd.DataFrame(columns=['sample_id', 'sentence', 'emotion', 'confidence', 'comments'])
+annotations = pd.DataFrame(columns=['sample_id', 'sentence', 'emotion', 'confidence', 'comments', 'n_clicks'])
 
 password_files = os.getenv("password_files")
 
@@ -267,7 +267,7 @@ def load_first_example(participant_id, ann_completed, 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)
+        annotations = pd.read_csv(path_ann, keep_default_na=False)
 
         current_index = len(annotations)
         print('path was found')
@@ -288,17 +288,17 @@ def load_example(index):
     # If the user already made an annotation for this example, gradio will return said annotation
     previous_annotation = (
         annotations.iloc[index].to_dict() if index < len(annotations) else {"sample_id": row["sample_id"], "emotion": 'Blank', "confidence": 0,
-                                                                            "comments": ''}
+                                                                            "comments": '', "n_clicks": 0}
     )
 
     start = row['start']
     end = row['end']
     duration = get_audio_duration(audio_path)
     print(f'{start} {end} {duration}')
-    return (sentence, audio_path, previous_annotation['emotion'], previous_annotation['confidence'], previous_annotation["comments"], start, end, duration)
+    return (sentence, audio_path, previous_annotation['emotion'], previous_annotation['confidence'], previous_annotation["comments"], n_clicks, start, end, duration)
 
 
-def save_annotation(emotions, confidence, comments, participant_id, ann_completed, current_index):
+def save_annotation(emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index):
     """Save the annotation for the current example."""
 
     row = file_list.iloc[current_index]
@@ -307,50 +307,50 @@ def save_annotation(emotions, confidence, comments, participant_id, ann_complete
 
     # Update or append annotation
     if sample_id in annotations["sample_id"].values:
-        annotations.loc[annotations["sample_id"] == sample_id, ["emotion", "confidence", "comments"]] = \
-            [emotions, confidence, comments]
+        annotations.loc[annotations["sample_id"] == sample_id, ["emotion", "confidence", "comments", "n_clicks"]] = \
+            [emotions, confidence, comments, n_clicks]
     else:
-        annotations.loc[len(annotations)] = [sample_id, sentence, emotions, confidence, comments]
+        annotations.loc[len(annotations)] = [sample_id, sentence, emotions, confidence, comments, n_clicks]
         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, ann_completed, current_index):
+def next_example(emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index):
     """Move to the next example."""
 
     if emotions == "Blank":
         gr.Warning("Please fill out the emotion section. 'Blank' is not a valid emotion.")
     else:
         
-        ann_completed = save_annotation(emotions, confidence, comments, participant_id, ann_completed, current_index)
+        ann_completed = save_annotation(emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index)
         if current_index < len(file_list) - 1:
             current_index += 1
     return *load_example(current_index), ann_completed, current_index
 
-def previous_example(emotion, confidence, comments, participant_id,  ann_completed, current_index):
+def previous_example(emotion, confidence, comments, n_clicks, participant_id,  ann_completed, current_index):
     """Move to the previous example."""
   
     if emotion != "Blank":
-        ann_completed = save_annotation(emotion, confidence, comments, participant_id,  ann_completed, current_index)
+        ann_completed = save_annotation(emotion, confidence, comments, n_clicks, participant_id,  ann_completed, current_index)
     if current_index > 0:
         current_index -= 1
             
     return *load_example(current_index), ann_completed, current_index
 
-def deactivate_participant_id(participant_id, lets_go, total, previous_button, next_button, sentence_text, audio_player, emotions, confidence, comments, ann_completed, current_index):
+def deactivate_participant_id(participant_id, lets_go, total, previous_button, next_button, sentence_text, audio_player, emotions, confidence, comments, n_clicks, ann_completed, current_index):
     global file_list
     global total_annotations
     if participant_id in possible_ids.keys():
-        file_list = pd.read_csv(os.path.join(persistent_storage, 'files_to_annotate_2round', f'group_{possible_ids[participant_id]}.csv'))
+        file_list = pd.read_csv(os.path.join(persistent_storage, 'files_to_annotate_2round', f'group_{possible_ids[participant_id]}.csv'), keep_default_na=False)
 
         
         
         total_annotations = len(file_list)
         total = gr.Number(total_annotations, visible=False)
+        
 
-
-        sentence, audio_player, emotions, confidence, comments, start, end, duration, ann_completed, current_index = load_first_example(participant_id, ann_completed, current_index)
+        sentence, audio_player, emotions, confidence, comments, n_clicks, start, end, duration, ann_completed, current_index = load_first_example(participant_id, ann_completed, current_index)
         
         participant_id = gr.Textbox(label='What is your participant ID?', value = participant_id, interactive = False)
         lets_go = gr.Button("Participant selected!", interactive = False)
@@ -362,8 +362,12 @@ def deactivate_participant_id(participant_id, lets_go, total, previous_button, n
 
     else:
         gr.Warning("Please insert a valid participant ID")
-    return participant_id, lets_go, total, previous_button, next_button, sentence_text, audio_player, emotions, confidence, comments, start, end, duration, ann_completed, current_index
+    return participant_id, lets_go, total, previous_button, next_button, sentence_text, audio_player, emotions, confidence, comments, n_clicks, start, end, duration, ann_completed, current_index
+
+def count_clicks(n_clicks):
+    n_clicks = gr.Number(n_clicks + 1, visible = False)
 
+    return n_clicks
 # ===================
 # Gradio Interface
 # ===================
@@ -408,6 +412,8 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         start = gr.Number(0, visible = False)
         end = gr.Number(0, visible = False)
         duration = gr.Number(0, visible = False)
+        n_clicks = gr.Number(0, visible = False)
+
         # Row with progress bar
         
         gr.HTML("""
@@ -446,19 +452,19 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         # Go back
         previous_button.click(
             previous_example,
-            inputs=[emotions, confidence, comments, participant_id,  ann_completed, current_index],
-            outputs=[sentence_text, audio_player, emotions, confidence, comments, start, end, duration, ann_completed, current_index],).then(None, [], [start, end, duration, current_index,ann_completed, total], js = js_progress_bar)
+            inputs=[emotions, confidence, comments, n_clicks, participant_id,  ann_completed, current_index],
+            outputs=[sentence_text, audio_player, emotions, confidence, comments, n_clicks, start, end, duration, ann_completed, current_index],).then(None, [], [start, end, duration, current_index,ann_completed, total], js = js_progress_bar)
 
         # Go to the next example
         next_button.click(
             next_example,
-            inputs=[emotions, confidence, comments, participant_id, ann_completed, current_index],
-            outputs=[sentence_text, audio_player, emotions, confidence, comments, start, end, duration, ann_completed, current_index],).then(None, [], [start, end, duration, current_index,ann_completed, total], js = js_progress_bar)
+            inputs=[emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index],
+            outputs=[sentence_text, audio_player, emotions, confidence, comments, n_clicks, start, end, duration, ann_completed, current_index],).then(None, [], [start, end, duration, current_index,ann_completed, total], js = js_progress_bar)
         
         buttons = [previous_button, next_button]
         data = [sentence_text, audio_player, emotions, confidence, comments]
-        lets_go.click(deactivate_participant_id, [participant_id, lets_go, total, *buttons, *data, ann_completed, current_index], [participant_id, lets_go, total, *buttons, *data, start, end, duration, ann_completed, current_index]).then( None, [], [start, end, duration, current_index, ann_completed, total], js = js_progress_bar)
-        
+        lets_go.click(deactivate_participant_id, [participant_id, lets_go, total, *buttons, *data, n_clicks, ann_completed, current_index], [participant_id, lets_go, total, *buttons, *data, n_clicks, start, end, duration, ann_completed, current_index]).then( None, [], [start, end, duration, current_index, ann_completed, total], js = js_progress_bar)
+        audio_player.play(count_clicks, [n_clicks], [n_clicks])
     
     with gr.Tab("Access Files"):
         with gr.Row():