Update app.py

app.py

@@ -1,93 +1,30 @@
+# app.py
+import os
 import gradio as gr
-from datasets import load_dataset
+from huggingface_hub import HfApi
 
-# Load dataset
-ds = load_dataset("Juna190825/ZomiAudioDataset", split="train")
+api = HfApi()
 
-# Build list of audio paths and mapping to full rows
-audio_paths = [row["audio"] for row in ds]
-audio_map = {row["audio"]: row for row in ds}
-
-def load_entry(index):
-    """Return audio + metadata for the given index."""
-    index = int(index)
-    path = audio_paths[index]
-    row = audio_map[path]
-
-    # Build metadata markdown
-    md = f"""
-### Metadata
-**Text:** {row.get("text", "")}
-
-**Speaker ID:** {row.get("speaker_id", "")}  
-**Gender:** {row.get("gender", "")}  
-**Age:** {row.get("age", "")}  
-**Accent:** {row.get("accent", "")}  
-**Genre:** {row.get("genre", "")}  
-
-**Start Time:** {row.get("start_time", "")}  
-**End Time:** {row.get("end_time", "")}  
-**Duration:** {row.get("duration", "")}  
-**Sampling Rate:** {row.get("sampling_rate", "")}  
-**Split:** {row.get("split", "")}
-"""
-
-    return path, md, index, path
-
-
-def next_audio(current_index):
-    """Move to the next audio file."""
-    new_index = (current_index + 1) % len(audio_paths)
-    return load_entry(new_index)
-
-
-def prev_audio(current_index):
-    """Move to the previous audio file."""
-    new_index = (current_index - 1) % len(audio_paths)
-    return load_entry(new_index)
-
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎧 Audio Dataset Browser with Navigation")
-
-    # State: current index
-    current_index = gr.State(0)
-
-    with gr.Row():
-        dropdown = gr.Dropdown(
-            choices=audio_paths,
-            label="Select an audio file",
-            value=audio_paths[0],
-            interactive=True
+def list_dataset_files(repo_id: str):
+    # repo_id example: "username/dataset_name"
+    try:
+        files = api.list_repo_files(
+            repo_id=repo_id,
+            repo_type="dataset",
+            token=os.getenv("HF_TOKEN")  # None for public datasets
         )
+        if not files:
+            return "No files found."
+        return "\n".join(files)
+    except Exception as e:
+        return f"Error: {e}"
 
-    with gr.Row():
-        player = gr.Audio(label="Audio Player")
-        metadata_box = gr.Markdown()
-
-    with gr.Row():
-        prev_btn = gr.Button("⬅️ Previous")
-        next_btn = gr.Button("Next ➡️")
-
-    # Dropdown selection updates everything
-    dropdown.change(
-        fn=lambda path: load_entry(audio_paths.index(path)),
-        inputs=dropdown,
-        outputs=[player, metadata_box, current_index, dropdown]
-    )
-
-    # Next button
-    next_btn.click(
-        fn=next_audio,
-        inputs=current_index,
-        outputs=[player, metadata_box, current_index, dropdown]
-    )
+with gr.Blocks() as demo:
+    gr.Markdown("# List files in a Hugging Face dataset repo")
+    repo_input = gr.Textbox(label="Dataset repo_id", value="username/dataset_name")
+    output = gr.Textbox(label="Files", lines=20)
+    btn = gr.Button("List files")
 
-    # Previous button
-    prev_btn.click(
-        fn=prev_audio,
-        inputs=current_index,
-        outputs=[player, metadata_box, current_index, dropdown]
-    )
+    btn.click(list_dataset_files, inputs=repo_input, outputs=output)
 
 demo.launch()