update

main.py

@@ -3,6 +3,8 @@
 import argparse
 from pathlib import Path
 import platform
+import shutil
+import zipfile
 
 import gradio as gr
 from huggingface_hub import snapshot_download
@@ -65,7 +67,7 @@ def when_click_denoise_button(noisy_audio_t, engine: str):
         raise gr.Error(f"enhancement failed, error type: {type(e)}, error text: {str(e)}.")
 
     enhanced_audio_t = (sample_rate, enhanced_audio)
-    return enhanced_audio_t, None
+    return enhanced_audio_t
 
 
 def main():
@@ -86,15 +88,33 @@ def main():
     # engines
     global denoise_engines
     denoise_engines = {
-        "mpnet": InferenceMPNet(
+        "mpnet_aishell_20250221": InferenceMPNet(
             pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet_aishell_20250221.zip").as_posix(),
         ),
-
     }
 
     # choices
     denoise_engine_choices = list(denoise_engines.keys())
 
+    # examples
+    example_zip_file = trained_model_dir / "examples.zip"
+    with zipfile.ZipFile(example_zip_file.as_posix(), "r") as f_zip:
+        out_root = examples_dir
+        if out_root.exists():
+            shutil.rmtree(out_root.as_posix())
+        out_root.mkdir(parents=True, exist_ok=True)
+        f_zip.extractall(path=out_root)
+
+    # examples
+    examples = list()
+    for filename in examples_dir.glob("**/*/*.wav"):
+        label = filename.parts[-2]
+
+        examples.append([
+            filename.as_posix(),
+            denoise_engine_choices[0]
+        ])
+
     # ui
     with gr.Blocks() as blocks:
         gr.Markdown(value="nx denoise.")
@@ -107,12 +127,11 @@ def main():
                         dn_button = gr.Button(variant="primary")
                     with gr.Column(variant="panel", scale=5):
                         dn_enhanced_audio = gr.Audio(label="enhanced_audio")
-                        dn_clean_audio = gr.Audio(label="clean_audio")
 
                 dn_button.click(
                     when_click_denoise_button,
                     inputs=[dn_noisy_audio, dn_engine],
-                    outputs=[dn_enhanced_audio, dn_clean_audio]
+                    outputs=[dn_enhanced_audio]
                 )
 
     # http://127.0.0.1:7864/