Update app.py

app.py

@@ -1,36 +1,9 @@
 # # app.py
-# import os
-# import gradio as gr
-# from huggingface_hub import HfApi
-
-# api = HfApi()
-
-# 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.Blocks() as demo:
-#     gr.Markdown("# List files in a Hugging Face dataset repo")
-#     repo_input = gr.Textbox(label="Dataset repo_id", value="Juna190825/ZomiAudioDataset")
-#     output = gr.Textbox(label="Files", lines=20)
-#     btn = gr.Button("List files")
-
-#     btn.click(list_dataset_files, inputs=repo_input, outputs=output)
-
-# demo.launch()
-
-
 import os
+import json
+import csv
+import tempfile
+import shutil
 import gradio as gr
 from huggingface_hub import HfApi, hf_hub_download
 
@@ -48,27 +21,68 @@ def list_dataset_files(repo_id: str):
         if not files:
             return [], "No files found."
 
-        # Filter audio files
         audio_files = [f for f in files if os.path.splitext(f)[1].lower() in AUDIO_EXTS]
-
         return audio_files, "\n".join(files)
 
     except Exception as e:
         return [], f"Error: {e}"
 
-# def load_audio(repo_id, file_path):
-#     try:
-#         local_path = hf_hub_download(
-#             repo_id=repo_id,
-#             filename=file_path,
-#             repo_type="dataset",
-#             token=os.getenv("HF_TOKEN")
-#         )
-#         return local_path
-#     except Exception as e:
-#         return None
-import shutil
-import tempfile
+
+# -----------------------------
+# METADATA LOADING
+# -----------------------------
+
+def load_metadata(repo_id):
+    """Download and parse metadata files if they exist."""
+    metadata = {}
+
+    # Possible metadata files
+    candidates = ["metadata.jsonl"] #, "metadata.json", "metadata.csv"]
+
+    for fname in candidates:
+        try:
+            path = hf_hub_download(
+                repo_id=repo_id,
+                filename=fname,
+                repo_type="dataset",
+                token=os.getenv("HF_TOKEN")
+            )
+
+            ext = os.path.splitext(fname)[1]
+
+            if ext == ".jsonl":
+                with open(path, "r", encoding="utf-8") as f:
+                    for line in f:
+                        item = json.loads(line)
+                        audio = item.get("audio") or item.get("file")
+                        if audio:
+                            metadata[audio] = item
+
+            elif ext == ".json":
+                with open(path, "r", encoding="utf-8") as f:
+                    data = json.load(f)
+                    for item in data:
+                        audio = item.get("audio") or item.get("file")
+                        if audio:
+                            metadata[audio] = item
+
+            elif ext == ".csv":
+                with open(path, "r", encoding="utf-8") as f:
+                    reader = csv.DictReader(f)
+                    for row in reader:
+                        audio = row.get("audio") or row.get("file")
+                        if audio:
+                            metadata[audio] = row
+
+        except Exception:
+            pass  # File doesn't exist, skip
+
+    return metadata
+
+
+# -----------------------------
+# AUDIO LOADING
+# -----------------------------
 
 def load_audio(repo_id, file_path):
     try:
@@ -79,19 +93,51 @@ def load_audio(repo_id, file_path):
             token=os.getenv("HF_TOKEN")
         )
 
-        # Copy to a temp file Gradio is allowed to serve
         tmp_dir = tempfile.gettempdir()
         safe_path = os.path.join(tmp_dir, os.path.basename(local_path))
         shutil.copy(local_path, safe_path)
 
         return safe_path
 
-    except Exception as e:
+    except Exception:
         return None
 
 
+# -----------------------------
+# COMBINED HANDLER
+# -----------------------------
+
+def load_audio_and_metadata(repo_id, file_path):
+    audio = load_audio(repo_id, file_path)
+    metadata = load_metadata(repo_id)
+
+    # Try to match metadata
+    info = metadata.get(file_path, "")
+
+    # If no metadata file, try matching .txt file
+    if not info:
+        txt_candidate = file_path.replace(".wav", ".txt").replace(".mp3", ".txt")
+        try:
+            txt_path = hf_hub_download(
+                repo_id=repo_id,
+                filename=txt_candidate,
+                repo_type="dataset",
+                token=os.getenv("HF_TOKEN")
+            )
+            with open(txt_path, "r", encoding="utf-8") as f:
+                info = f.read()
+        except Exception:
+            info = "No metadata found."
+
+    return audio, json.dumps(info, indent=2) if isinstance(info, dict) else info
+
+
+# -----------------------------
+# UI
+# -----------------------------
+
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎧 List & Play Audio Files from a HF Dataset Repo")
+    gr.Markdown("# 🎧 List, Play & View Metadata for HF Dataset Audio")
 
     repo_input = gr.Textbox(label="Dataset repo_id", value="username/dataset_name")
 
@@ -99,21 +145,21 @@ with gr.Blocks() as demo:
         file_list = gr.Dropdown(label="Audio files", choices=[])
         play_audio = gr.Audio(label="Audio Player")
 
+    metadata_box = gr.Textbox(label="Metadata / Text", lines=10)
     output = gr.Textbox(label="All Files", lines=20)
+
     btn = gr.Button("List files")
 
-    # When clicking "List files"
     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
 
     btn.click(update_files, inputs=repo_input, outputs=[file_list, output])
 
-    # When selecting an audio file
     file_list.change(
-        load_audio,
+        load_audio_and_metadata,
         inputs=[repo_input, file_list],
-        outputs=play_audio
+        outputs=[play_audio, metadata_box]
     )
 
 demo.launch()