Add previous and next button and autoplay toggle

app.py

@@ -1,4 +1,4 @@
-# # app.py
+# app.py
 import os
 import json
 import csv
@@ -108,6 +108,9 @@ def load_audio(repo_id, file_path):
 # -----------------------------
 
 def load_audio_and_metadata(repo_id, file_path):
+    if not file_path:
+        return None, "No file selected."
+    
     audio = load_audio(repo_id, file_path)
     metadata = load_metadata(repo_id)
 
@@ -132,6 +135,38 @@ def load_audio_and_metadata(repo_id, file_path):
     return audio, json.dumps(info, indent=2) if isinstance(info, dict) else info
 
 
+# -----------------------------
+# NAVIGATION FUNCTIONS
+# -----------------------------
+
+def navigate_files(current_file, direction, audio_files):
+    """Navigate to previous or next file in the list."""
+    if not audio_files or current_file not in audio_files:
+        return current_file
+    
+    current_index = audio_files.index(current_file)
+    
+    if direction == "next":
+        new_index = (current_index + 1) % len(audio_files)
+    elif direction == "prev":
+        new_index = (current_index - 1) % len(audio_files)
+    else:
+        return current_file
+    
+    return audio_files[new_index]
+
+
+def update_file_selection(repo_id, current_file, direction, audio_files_state, autoplay):
+    """Update file selection and load new audio/metadata."""
+    if not audio_files_state:
+        return current_file, None, "No audio files available."
+    
+    new_file = navigate_files(current_file, direction, audio_files_state)
+    audio, metadata = load_audio_and_metadata(repo_id, new_file)
+    
+    return new_file, audio, metadata
+
+
 # -----------------------------
 # UI
 # -----------------------------
@@ -140,10 +175,20 @@ with gr.Blocks() as demo:
     gr.Markdown("# 🎧 List, Play & View Metadata for HF Dataset Audio")
 
     repo_input = gr.Textbox(label="Dataset repo_id", value="username/dataset_name")
+    
+    # State to store audio files list
+    audio_files_state = gr.State([])
 
     with gr.Row():
-        file_list = gr.Dropdown(label="Audio files", choices=[])
-        play_audio = gr.Audio(label="Audio Player")
+        file_list = gr.Dropdown(label="Audio files", choices=[], interactive=True)
+        
+    with gr.Row():
+        prev_btn = gr.Button("◀ Previous")
+        next_btn = gr.Button("Next ▶")
+        autoplay_toggle = gr.Checkbox(label="Autoplay", value=True)
+    
+    with gr.Row():
+        play_audio = gr.Audio(label="Audio Player", autoplay=True)
 
     metadata_box = gr.Textbox(label="Metadata / Text", lines=10)
     output = gr.Textbox(label="All Files", lines=20)
@@ -152,14 +197,65 @@ with gr.Blocks() as demo:
 
     def update_files(repo_id):
         audio_files, all_files_text = list_dataset_files(repo_id)
-        return gr.Dropdown(choices=audio_files, value=None), all_files_text
+        # Update state and dropdown
+        return (
+            gr.Dropdown(choices=audio_files, value=audio_files[0] if audio_files else None),
+            all_files_text,
+            audio_files  # Update state
+        )
+
+    btn.click(
+        update_files, 
+        inputs=repo_input, 
+        outputs=[file_list, output, audio_files_state]
+    )
 
-    btn.click(update_files, inputs=repo_input, outputs=[file_list, output])
+    # When file is selected from dropdown
+    def dropdown_change_handler(repo_id, file_path, autoplay):
+        audio, metadata = load_audio_and_metadata(repo_id, file_path)
+        # Update audio player autoplay based on toggle
+        return gr.Audio(value=audio, autoplay=autoplay), metadata
 
     file_list.change(
-        load_audio_and_metadata,
-        inputs=[repo_input, file_list],
+        dropdown_change_handler,
+        inputs=[repo_input, file_list, autoplay_toggle],
         outputs=[play_audio, metadata_box]
     )
 
-demo.launch()
+    # Previous button click
+    def prev_button_handler(repo_id, current_file, audio_files_state, autoplay):
+        new_file, audio, metadata = update_file_selection(
+            repo_id, current_file, "prev", audio_files_state, autoplay
+        )
+        return new_file, gr.Audio(value=audio, autoplay=autoplay), metadata
+
+    prev_btn.click(
+        prev_button_handler,
+        inputs=[repo_input, file_list, audio_files_state, autoplay_toggle],
+        outputs=[file_list, play_audio, metadata_box]
+    )
+
+    # Next button click
+    def next_button_handler(repo_id, current_file, audio_files_state, autoplay):
+        new_file, audio, metadata = update_file_selection(
+            repo_id, current_file, "next", audio_files_state, autoplay
+        )
+        return new_file, gr.Audio(value=audio, autoplay=autoplay), metadata
+
+    next_btn.click(
+        next_button_handler,
+        inputs=[repo_input, file_list, audio_files_state, autoplay_toggle],
+        outputs=[file_list, play_audio, metadata_box]
+    )
+    
+    # Autoplay toggle change handler
+    def autoplay_changed(autoplay, current_audio):
+        return gr.Audio(value=current_audio, autoplay=autoplay)
+    
+    autoplay_toggle.change(
+        autoplay_changed,
+        inputs=[autoplay_toggle, play_audio],
+        outputs=[play_audio]
+    )
+
+demo.launch()