Add file examples

app.py

@@ -2,26 +2,38 @@ 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))
-current_index = 0
 
-# CSV to store annotations
 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)
 
-# Function to save annotation
-def save_annotation(user_id, gender, score):
-    global current_index
-    if current_index >= len(audio_files):
-        return "No more audio files.", None
-    
-    audio_file = audio_files[current_index]
-    
-    # Save to CSV
+# Main function (per-user session state)
+def load_next_audio(state):
+    if state is None:
+        state = {"index": 0}
+
+    idx = state["index"]
+
+    if idx >= len(audio_files):
+        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]}"
+
+def submit_annotation(user_id, gender, score, state):
+    if state is None:
+        state = {"index": 0}
+
+    idx = state["index"]
+
+    if idx >= len(audio_files):
+        return state, "No more audio files."
+
+    audio_file = audio_files[idx]
+
     df = pd.read_csv(ANNOTATION_FILE)
     df = pd.concat([df, pd.DataFrame([{
         "user_id": user_id,
@@ -30,33 +42,40 @@ def save_annotation(user_id, gender, score):
         "score": score
     }])], ignore_index=True)
     df.to_csv(ANNOTATION_FILE, index=False)
-    
-    current_index += 1
-    if current_index < len(audio_files):
-        next_audio = os.path.join(AUDIO_FOLDER, audio_files[current_index])
-    else:
-        next_audio = None
-    
-    return "Saved! Next audio loaded." if next_audio else "All audios completed.", next_audio
-
-# Gradio interface
+
+    state["index"] += 1
+    return state, f"Saved rating for {audio_file}."
+
 with gr.Blocks() as demo:
+    gr.Markdown("# Audio MOS Annotation Tool")
+
+    state = gr.State({"index": 0})
+
     with gr.Row():
         user_id = gr.Textbox(label="User ID")
-        gender = gr.Dropdown(label="Gender", choices=["Male", "Female", "Other"])
-    audio_player = gr.Audio(label="Audio File", source="upload", type="filepath")
-    score = gr.Slider(label="MOS Score (1-5)", minimum=1, maximum=5, step=1)
+        gender = gr.Dropdown(["Male", "Female", "Other"], label="Gender")
+
+    audio_player = gr.Audio(label="Audio File", autoplay=False)
+    score = gr.Slider(label="MOS Score (1–5)", minimum=1, maximum=5, step=1)
+
     submit_btn = gr.Button("Submit Score")
-    status = gr.Textbox(label="Status")
-    
+    status = gr.Textbox(label="Status", interactive=False)
+
+    # Load audio on startup
+    demo.load(load_next_audio, inputs=state,
+              outputs=[audio_player, state, status])
+
     submit_btn.click(
-        fn=save_annotation,
-        inputs=[user_id, gender, score],
-        outputs=[status, audio_player]
+        submit_annotation,
+        inputs=[user_id, gender, score, state],
+        outputs=[state, status],
     )
 
-# Load first audio
-audio_player.value = os.path.join(AUDIO_FOLDER, audio_files[current_index])
+    # After submitting, load next audio
+    submit_btn.click(
+        load_next_audio,
+        inputs=state,
+        outputs=[audio_player, state, status],
+    )
 
 demo.launch()
-