Download

annotations.csv

@@ -0,0 +1,3 @@
+user_id,gender,audio_file,score
+,[],audio1.wav,3
+,[],audio2.wav,3

app.py

@@ -2,15 +2,17 @@ import gradio as gr
 import pandas as pd
 import os
 
+# Folder with your audio files
 AUDIO_FOLDER = "audio_files"
 audio_files = sorted(os.listdir(AUDIO_FOLDER))
 
+# Annotation file stored in persistent space storage
 ANNOTATION_FILE = "annotations.csv"
 if not os.path.exists(ANNOTATION_FILE):
     df = pd.DataFrame(columns=["user_id", "gender", "audio_file", "score"])
     df.to_csv(ANNOTATION_FILE, index=False)
 
-# Main function (per-user session state)
+# Load next audio per user session
 def load_next_audio(state):
     if state is None:
         state = {"index": 0}
@@ -21,8 +23,9 @@ def load_next_audio(state):
         return None, state, "All audio files annotated."
 
     filepath = os.path.join(AUDIO_FOLDER, audio_files[idx])
-    return filepath, state, f"Loaded file {audio_files[idx]}"
+    return filepath, state, f"Loaded {audio_files[idx]}"
 
+# Save rating
 def submit_annotation(user_id, gender, score, state):
     if state is None:
         state = {"index": 0}
@@ -34,6 +37,7 @@ def submit_annotation(user_id, gender, score, state):
 
     audio_file = audio_files[idx]
 
+    # Append to CSV
     df = pd.read_csv(ANNOTATION_FILE)
     df = pd.concat([df, pd.DataFrame([{
         "user_id": user_id,
@@ -43,39 +47,57 @@ def submit_annotation(user_id, gender, score, state):
     }])], ignore_index=True)
     df.to_csv(ANNOTATION_FILE, index=False)
 
+    # Move to next audio
     state["index"] += 1
+
     return state, f"Saved rating for {audio_file}."
 
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Audio MOS Annotation Tool")
 
+    # Per-user session state
     state = gr.State({"index": 0})
 
+    # User registration
     with gr.Row():
         user_id = gr.Textbox(label="User ID")
         gender = gr.Dropdown(["Male", "Female", "Other"], label="Gender")
 
+    # Audio playback + rating
     audio_player = gr.Audio(label="Audio File", autoplay=False)
-    score = gr.Slider(label="MOS Score (1–5)", minimum=1, maximum=5, step=1)
+    score = gr.Slider(minimum=1, maximum=5, step=1, label="MOS Score (1–5)")
 
     submit_btn = gr.Button("Submit Score")
     status = gr.Textbox(label="Status", interactive=False)
 
-    # Load audio on startup
-    demo.load(load_next_audio, inputs=state,
-              outputs=[audio_player, state, status])
+    # Download annotations.csv
+    download_btn = gr.File(label="Download annotations.csv")
+
+    # Load first audio automatically
+    demo.load(
+        load_next_audio,
+        inputs=state,
+        outputs=[audio_player, state, status]
+    )
 
+    # When submitting a score:
+    # 1) Save annotation
     submit_btn.click(
         submit_annotation,
         inputs=[user_id, gender, score, state],
         outputs=[state, status],
     )
 
-    # After submitting, load next audio
+    # 2) Load next audio
     submit_btn.click(
         load_next_audio,
         inputs=state,
         outputs=[audio_player, state, status],
     )
 
+    # Always update the download button with current CSV
+    demo.load(lambda: ANNOTATION_FILE, None, download_btn)
+    submit_btn.click(lambda: ANNOTATION_FILE, None, download_btn)
+
 demo.launch()