Dasheng view: dropdown only shows batch_outputs_dasheng sample IDs (no fold*)

app.py

@@ -9,12 +9,14 @@ import gradio as gr
 from dataset_loader import (
     DASHENG_PREFIX,
     list_samples,
+    list_samples_dasheng,
     get_nn_demo_paths,
     get_results_demo_paths,
 )
 
 
 SAMPLE_IDS = list_samples()
+DASHENG_SAMPLE_IDS = list_samples_dasheng()  # Only IDs in batch_outputs_dasheng (no fold*)
 TOP1_ID = SAMPLE_IDS[0] if SAMPLE_IDS else None
 
 
@@ -130,16 +132,26 @@ with gr.Blocks(title="NearestNeighbor Audio Demo", css=".gradio-container { max-
 
     def on_change(sid, view):
         use_dasheng = view == "Nearest Neighbor (Dasheng)"
+        # Dasheng view: only show IDs that exist in batch_outputs_dasheng (no fold*)
+        if use_dasheng:
+            choices = DASHENG_SAMPLE_IDS
+            if sid not in DASHENG_SAMPLE_IDS and DASHENG_SAMPLE_IDS:
+                sid = DASHENG_SAMPLE_IDS[0]
+        else:
+            choices = SAMPLE_IDS
+            if sid not in SAMPLE_IDS and SAMPLE_IDS:
+                sid = SAMPLE_IDS[0]
         nn_vals = build_nn_view(sid, use_dasheng=use_dasheng)
         res_vals = build_results_view(sid)
         is_nn = view in ("Nearest Neighbor (CLAP)", "Nearest Neighbor (Dasheng)")
         nn_title = "### Nearest Neighbor (Dasheng): Baseline outputs (top 10 prompts)" if use_dasheng else "### Nearest Neighbor (CLAP): Baseline outputs (top 10 prompts)"
+        dd_update = gr.update(choices=choices, value=sid)
         return (
             [gr.update(value=nn_title)] + list(nn_vals) + list(res_vals) +
-            [gr.update(visible=is_nn), gr.update(visible=not is_nn)]
+            [gr.update(visible=is_nn), gr.update(visible=not is_nn), dd_update]
         )
 
-    all_outputs = [nn_section_title] + nn_outputs + res_outputs + [nn_col, res_col]
+    all_outputs = [nn_section_title] + nn_outputs + res_outputs + [nn_col, res_col, noise_dd]
 
     noise_dd.change(on_change, inputs=[noise_dd, view_radio], outputs=all_outputs)
     view_radio.change(on_change, inputs=[noise_dd, view_radio], outputs=all_outputs)

dataset_loader.py

@@ -61,6 +61,11 @@ def list_samples() -> list[str]:
     return _get_all_sample_ids()
 
 
+def list_samples_dasheng() -> list[str]:
+    """Return only sample IDs that exist in batch_outputs_dasheng (no fold* from UrbanSound8k)."""
+    return _get_sample_ids(DASHENG_PREFIX)
+
+
 def _find_files(inner: str) -> list[str]:
     """List all repo files under inner path."""
     files = list_repo_files(REPO_ID, repo_type=REPO_TYPE)