update

main.py

@@ -30,9 +30,13 @@ def get_args():
     args = parser.parse_args()
     return args
 
-def process_uploaded_file(filename: str, silence_time: float = 0.3, longest_activate: float = 3.0) -> str:
-    return run_nx_vad(filename, silence_time, longest_activate)
-
+def process_uploaded_file(vad_engine: str, filename: str, silence_time: float = 0.3, longest_activate: float = 3.0) -> str:
+    if vad_engine == "nx_vad":
+        return run_nx_vad(filename, silence_time, longest_activate)
+    elif vad_engine == "silero-vad":
+        return "silero-vad"
+    else:
+        return f"vad engine invalid: {vad_engine}"
 
 def run_nx_vad(filename: str, silence_time: float = 0.3, longest_activate: float = 3.0) -> str:
     filename = Path(filename).as_posix()
@@ -48,7 +52,7 @@ def run_nx_vad(filename: str, silence_time: float = 0.3, longest_activate: float
 
     vad_timestamps = re.findall(pattern, vad_result, flags=re.DOTALL)
 
-    vad_timestamps: str = json.dumps(vad_timestamps, ensure_ascii=False)
+    vad_timestamps: str = json.dumps(vad_timestamps, ensure_ascii=False, indent=2)
 
     return vad_timestamps
 
@@ -69,6 +73,7 @@ def main():
     for filename in example_wav_dir.glob("*.wav"):
         examples.append(
             [
+                "nx_vad",
                 filename.as_posix(),
                 0.3, 3.0,
             ]
@@ -85,6 +90,7 @@ def main():
                     type="filepath",
                     label="Upload from disk",
                 )
+                uploaded_vad_engine = gr.Dropdown(choices=["nx_vad", "silero-vad"], value="nx_vad", label="vad_engine")
                 uploaded_silence_time = gr.Slider(minimum=0.0, maximum=1.0, value=0.3, step=0.01, label="silence time")
                 uploaded_longest_activate = gr.Slider(minimum=0.0, maximum=20.0, value=3.0, step=0.1, label="longest activate")
                 upload_button = gr.Button("Run VAD")
@@ -93,6 +99,7 @@ def main():
             gr.Examples(
                 examples=examples,
                 inputs=[
+                    uploaded_vad_engine,
                     uploaded_file,
                     uploaded_silence_time,
                     uploaded_longest_activate,
@@ -106,6 +113,7 @@ def main():
             upload_button.click(
                 process_uploaded_file,
                 inputs=[
+                    uploaded_vad_engine,
                     uploaded_file,
                     uploaded_silence_time,
                     uploaded_longest_activate,