Feat: add progress bar

annotations.csv

@@ -1,2 +1,5 @@
 sample_id,sentence,emotion,comments
-007-0023,I was mostly hanging out in East Village in the clubs.,Angry,
+007-0023,I was mostly hanging out in East Village in the clubs.,Neutral,
+007-0034,"And we became friends, and then he told me that his lover had died of AIDS.",Angry,
+007-0086,"I mean, given the size of ACT UP, I mean, really, you're talking about a handful.",Neutral,
+007-0129,"And why is he like, you know, why are you putting him in yellow face?",Neutral,

app.py

@@ -6,13 +6,18 @@ import gradio as gr
 
 # List of all audio files to annotate
 file_list = pd.read_excel(os.path.join('combined_annotations.xlsx'))
+annotations_completed = 0
+total_annotations = len(file_list)
 
 # Initialize an empty DataFrame to store annotations
 annotations = pd.DataFrame(columns=['sample_id', 'sentence', 'emotion', 'comments'])
 current_index = {"index": 0}  # Dictionary to allow modifying inside functions
 
+
 def load_example(index):
-    """Load the example (audio + text) by 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')
     print(f"Audio path: {audio_path}, Exists: {os.path.exists(audio_path)}")
@@ -41,18 +46,19 @@ def save_annotation(emotions, comments):
         annotations.loc[len(annotations)] = [sample_id, sentence, emotions, comments]
 
     annotations.to_csv("annotations.csv", index=False)  # Save to a CSV file
-    #return f"Saved annotations for example {idx + 1}"
+    
 
 def next_example(emotions, comments):
+    """Move to the next example."""
     if emotions == '':
         raise gr.Error("Please fill out the emotion section")
     else:
-        """Move to the next example."""
+        
         save_annotation(emotions, comments)
-
         if current_index["index"] < len(file_list) - 1:
             current_index["index"] += 1
             return load_example(current_index["index"])
+
     #return "End of examples", None, 0, 0, 0, 0, ''
 
 
@@ -64,12 +70,21 @@ def previous_example(emotion, comments):
         return load_example(current_index["index"])
     return load_example(current_index["index"])
 
-# Gradio Interface
+def update_progress():
+    """Update annotation progress"""
+    global annotations_completed
+    if annotations_completed < total_annotations:
+        annotations_completed += 1  
+    
+    progress = annotations_completed / total_annotations
+    return f"Progress: {annotations_completed}/{annotations_tasks}"
 
+# Gradio Interface
 audio_path = 'test.mp3'
-with (gr.Blocks() as demo):
-    
-    
+with (gr.Blocks(theme=gr.themes.Soft()) as demo):
+    gr.Markdown("# Task Progress Tracker")
+    progress_text = gr.Textbox(value="Progress: 0/10", interactive=False)
+
     # Row with audio player
     with gr.Row():
         audio_player = gr.Audio(value=audio_path, label="Audio", type="filepath", interactive=False)
@@ -87,29 +102,41 @@ with (gr.Blocks() as demo):
 
     # Instructions for emotion annotation
     with gr.Sidebar():
-        gr.Textbox()
+        happy_words = gr.Textbox(label = "Happy")
     
 
     # Next and Previous Buttons
     with gr.Row():
-        next_button = gr.Button("Next Example")
         previous_button = gr.Button("Previous Example")
+        next_button = gr.Button("Next Example")
+
 
     comments = gr.Textbox(label="Comments", interactive=True)
-    # Initial load
+
+ 
     sentence_text.value, audio_player.value, emotions.value, comments.value = load_example(
-        current_index["index"]
-    )
+        current_index["index"])
 
-    next_button.click(
-        next_example,
+    previous_button.click(
+        previous_example,
         inputs=[emotions, comments],
         outputs=[sentence_text, audio_player, emotions, comments],
     )
-    previous_button.click(
-        previous_example,
+
+    next_button.click(
+        next_example,
         inputs=[emotions, comments],
         outputs=[sentence_text, audio_player, emotions, comments],
     )
 
+    next_button.click(update_progress,  outputs=[progress_text])
+
+    
+
+    
+
+
+    
+    
 demo.launch()
+