remove option for noise adding

app.py

@@ -1,5 +1,4 @@
 import os
-from random import sample
 import time
 from typing import Optional, Any
 
@@ -7,9 +6,6 @@ import gradio as gr
 from loguru import logger
 
 from constants import (
-    NOISES,
-    SNR_LEVELS,
-    NOISE_TYPES,
     MINUTES_KEEP,
 )
 from sdk import SDKParams, process_file_sdk
@@ -53,8 +49,6 @@ def cleanup_tmp(minutes_keep: int = MINUTES_KEEP, filter: list[str] = []):
 
 def denoise_audio(
     sample_path: str,
-    noise_type: str,
-    snr: str,
     enhancement_level: float = 50.0,
 ) -> tuple[Optional[str], Optional[str], Optional[str], Optional[str]]:
     gr.Info(
@@ -65,17 +59,7 @@ def denoise_audio(
     noisy_path = f"{base}_noisy{ext}"
     noisy_spec_path = f"{base}_noisy_spectrogram.png"
     enhanced_spec_path = f"{base}_enhanced_spectrogram.png"
-    if noise_type != "None":
-        clipped = mix_at_snr(
-            signal_path=sample_path,
-            noise_path=NOISES[noise_type],
-            output_path=noisy_path,
-            snr_db=int(snr),
-        )
-        if clipped:
-            gr.Warning("Adding noise caused clipping. Normalizing might alter the SNR.")
-    else:
-        noisy_path = sample_path
+    noisy_path = sample_path
     try:
         sdk_key = os.getenv("SECRET_SDK_KEY")
         sdk_params = SDKParams(
@@ -190,17 +174,6 @@ with gr.Blocks() as demo:
             audio_file = gr.Audio(
                 type="filepath", label="Input", visible=True, sources=["upload", "microphone"]
             )
-            noise_type = gr.Dropdown(
-                label="Add noise",
-                choices=[*NOISE_TYPES],
-                value="None",
-            )
-            noise_level = gr.Dropdown(
-                label="Noise Level (SNR)",
-                choices=[*SNR_LEVELS],
-                value="None",
-                visible=False,
-            )
             percent_slider = gr.Slider(
                 minimum=1,
                 maximum=100,
@@ -228,13 +201,10 @@ with gr.Blocks() as demo:
         denoise_audio,
         inputs=[
             input_enhancement,
-            noise_type,
-            noise_level,
             percent_slider,
         ],
         outputs=[noisy_audio, noisy_image, enhanced_audio, enhanced_image],
     ).then(enable_new_input, None, btn)
-    noise_type.change(toggle_SNR, noise_type, noise_level)
 
 cleanup_tmp(minutes_keep=0, filter=[])
 demo.launch(allowed_paths=["/tmp", "/"])