Feat: Restrict participant ids

app.py

@@ -210,13 +210,19 @@ side_bar_html = """
 
 persistent_storage = Path('/data')
 # List of all audio files to annotate
-file_list = pd.read_excel(os.path.join('combined_annotations.xlsx'))
-total_annotations = len(file_list)
+global file_list
+global total_annotations 
 # 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
 password_files = os.getenv("password_files")
 
+possible_ids = {'Tiger-001': 0, 'Panda-002': 0, 
+                'Falcon-003': 1, 'Wolf-004': 1,
+                'Dolphin-005': 2, 'Eagle-006': 2,
+                'Jaguar-007': 3, 'Rhino-008': 3,
+                'Zebra-009': 4, 'Lion-010': 4,
+                'Cheetah-011': 5, 'Bear-012': 5}
 
 def get_storage(password):
     #source: https://discuss.huggingface.co/t/accessing-data-folder-of-persistent-storage/46681/2
@@ -238,11 +244,10 @@ def get_storage(password):
     return files, f"{usage/(1024.0 ** 3):.3f}GB"
 
 
-def load_first_example(index, participant_id, ann_completed):
+def load_first_example(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):
@@ -255,18 +260,18 @@ def load_first_example(index, participant_id, ann_completed):
         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. 
     If there are any annotations it will give those values to the annotation dataframe"""
     
     row = file_list.iloc[index]
-    audio_path = os.path.join('files_to_annotate_padded_smaller_emotion_set', row["SAMPLE ID"].split('-')[0], row["SAMPLE ID"] + '.wav')
-    sentence = row["SENTENCE"]
+    audio_path = os.path.join(persistent_storage, 'files_to_annotate_2round', row["sample_id"].split('-')[0], row["sample_id"] + '.wav')
+    sentence = row["sentence"]
 
     # 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,
+        annotations.iloc[index].to_dict() if index < len(annotations) else {"sample_id": row["sample_id"], "emotion": 'Blank', "confidence": 0,
                                                                             "comments": ''}
     )
     print(previous_annotation)
@@ -278,8 +283,8 @@ def save_annotation(emotions, confidence, comments, participant_id, ann_complete
 
     idx = current_index["index"]
     row = file_list.iloc[idx]
-    sample_id = row["SAMPLE ID"]
-    sentence = row["SENTENCE"]
+    sample_id = row["sample_id"]
+    sentence = row["sentence"]
 
     # Update or append annotation
     if sample_id in annotations["sample_id"].values:
@@ -307,27 +312,38 @@ def next_example(emotions, confidence, comments, participant_id, ann_completed):
 def previous_example(emotion, confidence, comments, participant_id,  ann_completed):
     """Move to the previous example."""
     global current_index
-    if emotion.value != "Blank":
+    if emotion != "Blank":
         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"])
+    if current_index["index"] > 0:
+        current_index["index"] -= 1
+            
+    return 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)
-    lets_go = gr.Button("Participant selected!", interactive = False)
-    return participant_id, lets_go
+def deactivate_participant_id(participant_id, lets_go, total, previous_button, next_button, sentence_text, audio_player, emotions, confidence, comments, ann_completed):
+    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'))
 
-def activate_elements(emotions, confidence, comments, next_button, previous_button):
+        participant_id = gr.Textbox(label='What is your participant ID?', value = participant_id, interactive = False)
+        lets_go = gr.Button("Participant selected!", interactive = False)
+        
+        total_annotations = len(file_list)
+        total = gr.Number(total_annotations, visible=False)
 
-    emotions = gr.Radio(["Blank", "Joy", "Sad", "Angry", "Neutral"], label="Predominant Emotion", value = "Blank", visible = True)
-    confidence = gr.Slider(label="How confident are you that the annotated emotion is present in the recording? (%)", minimum=0, maximum=100, step=10, visible = True)
-    comments = gr.Textbox(label="Comments", visible =True)
-    previous_button = gr.Button("Previous Example", visible = True)
-    next_button = gr.Button("Next Example",visible = True)
 
-    return emotions, confidence, comments, next_button, previous_button
+        ann_completed, sentence, audio_player, emotions, confidence, current_index['index'], comments = load_first_example(participant_id, ann_completed)
+        
+        emotions = gr.Radio(["Blank", "Joy", "Sad", "Angry", "Neutral"], label="Predominant Emotion", value =  emotions, visible = True)
+        confidence = gr.Slider(label="How confident are you that the annotated emotion is present in the recording? (%)", minimum=0, maximum=100, step=10, visible = True, value = confidence)
+        comments = gr.Textbox(label="Comments", visible =True, value = comments)
+        previous_button = gr.Button("Previous Example", visible = True)
+        next_button = gr.Button("Next Example",visible = True)
+
+    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, ann_completed
+
 # ===================
 # Gradio Interface
 # ===================
@@ -367,7 +383,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)
+        total = gr.Number(0, visible=False)
         # Row with progress bar
         
         gr.HTML("""
@@ -416,13 +432,11 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
             inputs=[emotions, confidence, comments, participant_id, ann_completed],
             outputs=[sentence_text, audio_player, emotions, confidence, ann_completed, comments],
         ).then(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, 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)
+        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], [participant_id, lets_go, total, *buttons, *data, ann_completed]).then( None, [], [ann_completed, total], js = js_progress_bar)
+        
     
     with gr.Tab("Access Files"):
         with gr.Row():
@@ -436,5 +450,5 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         get_files_button.click(get_storage, inputs= [password], outputs=[files, storage], postprocess=False)
 
         
-demo.launch()
+demo.launch(allowed_paths = ['/data/files_to_annotate_2round'])
 

files_to_annotate_padded_smaller_emotion_set/002/002-0035.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:14c71ba98601d6a8bfc20e354ccde13275391a811a243ad0d5182d86659ab293
-size 198956

files_to_annotate_padded_smaller_emotion_set/002/002-0061.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:df234dc09991531122d303b12915cb344b9e1d83ed17e994c58ff261c744e2a5
-size 203708

files_to_annotate_padded_smaller_emotion_set/002/002-0070.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:99921e1d55902b9f29c66fca97f6713033a1f57377d1e4febf08d7d462d585c8
-size 183552

files_to_annotate_padded_smaller_emotion_set/002/002-0076.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9bf2bb7a826622bed62013cd7bf163176c474a501284fa43c743344d1c0172ad
-size 282428

files_to_annotate_padded_smaller_emotion_set/002/002-0080.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:28776f505414e937277e60f71632ed179b52c741fe37c2e80389d346afab3ba1
-size 204620

files_to_annotate_padded_smaller_emotion_set/002/002-0081.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4c3c93632a4ebcc40d7f03d95c24b4612e8d19e9c4f6c2b97a7106f0938d4af6
-size 108576

files_to_annotate_padded_smaller_emotion_set/002/002-0107.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e5087f893589b22fa4d0305a623abf9a4bc44f7d5a7396890737411df83ae39a
-size 132620

files_to_annotate_padded_smaller_emotion_set/002/002-0152.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bfb40acadcb838abf688ab815137b5d7a8dcd055564e59eb40d4de56ca3a2911
-size 249882

files_to_annotate_padded_smaller_emotion_set/002/002-0155.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e1cab437cd33680b552801264483a2562e73c676542c5e348d64b1425173b33c
-size 447836

files_to_annotate_padded_smaller_emotion_set/002/002-0165.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2aa8d0ee040bad3b69432de5db7b304e5b51b2dcd2bc17021b3931798f2cecac
-size 170106

files_to_annotate_padded_smaller_emotion_set/002/002-0196.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1dd3945892c7bb8c8ce983a638f9c2f2865706152cfffd9c4913afb20f57040f
-size 145100

files_to_annotate_padded_smaller_emotion_set/002/002-0197.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1efecfb8e3291d344d24ebba92c0c9a0239808b8b1e90ee98f0b4f0e9321046d
-size 507212

files_to_annotate_padded_smaller_emotion_set/002/002-0210.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4cc7b5f40c1e5fb56bcc540df7e73b459dd7873e4ff1a34925be20fc7b486592
-size 167180

files_to_annotate_padded_smaller_emotion_set/002/002-0227.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:737a7e4b38050bbf9bb13c0a45e9cc0da0c48f206667d52786f4b6c02c5e114a
-size 128924

files_to_annotate_padded_smaller_emotion_set/002/002-0237.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7fc061371b0cf2d4c1418880527f6cb17504ff9c87494b33c3b9ebb5c9e0eeaa
-size 262316

files_to_annotate_padded_smaller_emotion_set/002/002-0256.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6223cf7419aad597dc435647fb255ba03f2576b566cb579c2325878e7497ce5c
-size 255594

files_to_annotate_padded_smaller_emotion_set/002/002-0257.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:04808f0449a0b869dcd3ec8b08577ef7b2656675bb884bdbd62305cc53ff2a6c
-size 273834

files_to_annotate_padded_smaller_emotion_set/002/002-0261.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0e51b19514c981aeaff85180ff3b6e5935f81cc1cdf85d03215d894553859fdf
-size 285404

files_to_annotate_padded_smaller_emotion_set/002/002-0320.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5fcfc22e695ba85912b3d2fc1b3ece5988902720dfe838b8fed47f21a0124ec1
-size 113418

files_to_annotate_padded_smaller_emotion_set/002/002-0346.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e2b7b601a1ff4156def2ed2badfc388345fc5c141e821f2ae7150a8c4d5cfbba
-size 116252

files_to_annotate_padded_smaller_emotion_set/002/002-0417.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:983972b15e03f4a1873e28d24a680d73664a1e11209c2e55c4d45d094844faaf
-size 146972

files_to_annotate_padded_smaller_emotion_set/002/002-0418.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0ad6e0d7963d51de31f465246347b69521b72cb89e2968d50e82e2e84f7a18d8
-size 135452

files_to_annotate_padded_smaller_emotion_set/002/002-0419.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0075b1fbc6a20d6c807d8e0c1c56608be61e9b01a59ec2fd6db2bc498b8b2c2c
-size 158588

files_to_annotate_padded_smaller_emotion_set/002/002-0424.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:149a45880fdac3aa6f7b0b4dd2e879a049233d2390e47920375137ded0f9dd10
-size 126816

files_to_annotate_padded_smaller_emotion_set/002/002-0431.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4e92bd58cad0f880b2621bb9497adcb0d0455ba1b9e64126a3b2576d4ac68d90
-size 151872

files_to_annotate_padded_smaller_emotion_set/002/002-0448.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:07350c2223751f28a14528eb0c25c8782d5f888a1fce947662cd215627469699
-size 203712

files_to_annotate_padded_smaller_emotion_set/002/002-0464.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:73787c593f0b56495812a820518391623a2bd8e80da5909f52975ea323cf13c0
-size 120140

files_to_annotate_padded_smaller_emotion_set/002/002-0504.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9f40f61253670571f85c9a0ce0b07116733fea7633eae5faadf84e23ec624e9c
-size 232512

files_to_annotate_padded_smaller_emotion_set/002/002-0513.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:05cd90cacd8134a73617ae6ef7194d0635cc918e67407339d6dd395c4a59559c
-size 103772

files_to_annotate_padded_smaller_emotion_set/002/002-0522.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c8c2ceeed2460d2e52e46f8d208d88ab040fc3cc81e29982cc2c8518a6292565
-size 322700

files_to_annotate_padded_smaller_emotion_set/002/002-0526.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7336f827941d9ecbd23f48b54b7de3d226af7983038c6a6b0f96de842034ba5e
-size 118220

files_to_annotate_padded_smaller_emotion_set/002/002-0530.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a080b3c7a362b548ba7f631b980bf75ee13ff9eed3a2421f9907473c1e8fae95
-size 174860

files_to_annotate_padded_smaller_emotion_set/002/002-0562.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0925fd9bdaedbab7a0b3409324c8d6a8344cde878699d3dd34932f1cdd295c69
-size 204764

files_to_annotate_padded_smaller_emotion_set/002/002-0573.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:105c851a83a1e1e9452f9650df961a8b89129238551eed8b46c73573d9e79791
-size 116204

files_to_annotate_padded_smaller_emotion_set/002/002-0581.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7fd110ac0ffa2afd53eabea39538e9271be233333c41371608a67e9597ae9eac
-size 167180

files_to_annotate_padded_smaller_emotion_set/002/002-0589.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a37cfb591d608e5215012d4c0e427b95f9eb94e233c71e037b1250f42567c5c6
-size 112412

files_to_annotate_padded_smaller_emotion_set/002/002-0598.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:576cf1c0abcc08850aa0cf58f12f5c511bf393b6b40b7818baab29195f7621bf
-size 148940

files_to_annotate_padded_smaller_emotion_set/002/002-0603.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8e140b1f3a3fb2921070b51f04f21e8cf49de288eeb33ee6e5154f1d75a29a31
-size 128736

files_to_annotate_padded_smaller_emotion_set/002/002-0610.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:45bbbf1c7c1b66d0cac25f8ec9f58f9ff5436458056b662e7e734cc7e4665907
-size 357308

files_to_annotate_padded_smaller_emotion_set/002/002-0611.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0384553af653ae9f43e4fd2e7817d14cf55570df5db4866bdcd56f4420ca3ef6
-size 142172

files_to_annotate_padded_smaller_emotion_set/002/002-0615.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:01ccecbeacbfed2d0f3ace3c40fba9925955cab382489dc40cf715b2bb530f39
-size 161424

files_to_annotate_padded_smaller_emotion_set/002/002-0624.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:acc38e184d4c4ffa3456d73c426a4b682b11b730bec9561bfc29a63dcf02e98c
-size 215180

files_to_annotate_padded_smaller_emotion_set/002/002-0645.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:910e222e8b9b63e39eab06cf572dd9deb0b66159ca77a6d3f115a245f76320ed
-size 141164

files_to_annotate_padded_smaller_emotion_set/002/002-0646.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:881c8f60252e74a7d0146a2c1a5b31bfea2096e941b41297fa6c694865ac87f7
-size 235296