Update app.py

app.py

@@ -3,9 +3,8 @@ import soundfile as sf
 from scipy import signal
 import gradio as gr
 
-def generate_vinyl_sound(noise_ratio, lowcut, highcut):
+def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     # Parameters
-    duration = 10.0  # duration of the sound in seconds
     sample_rate = 44100  # sample rate in Hz
     num_samples = int(duration * sample_rate)
 
@@ -32,7 +31,6 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut):
     hiss_noise = signal.lfilter(b, a, hiss_noise)
 
     # Generate pops
-    pop_rate = 10  # average number of pops per second
     num_pops = int(duration * pop_rate)
     pop_times = np.random.randint(0, num_samples, num_pops)
     pop_data = np.zeros(num_samples)
@@ -53,18 +51,20 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut):
 iface = gr.Interface(
     generate_vinyl_sound,
     [
-        gr.inputs.Slider(minimum=0, maximum=1, default=0.7, step=0.1, label="Noise Ratio"),
+        gr.inputs.Slider(minimum=0, maximum=1, default=0.001, step=0.001, label="Noise Ratio"),
         gr.inputs.Slider(minimum=20, maximum=20000, default=300, step=10, label="Lowcut Frequency (Hz)"),
-        gr.inputs.Slider(minimum=20, maximum=20000, default=5000, step=10, label="Highcut Frequency (Hz)")
+        gr.inputs.Slider(minimum=20, maximum=20000, default=5000, step=10, label="Highcut Frequency (Hz)"),
+        gr.inputs.Slider(minimum=1, maximum=60, default=10, step=1, label="Duration (seconds)"),
+        gr.inputs.Slider(minimum=1, maximum=100, default=10, step=1, label="Pop Rate (pops per second)")
     ],
     gr.outputs.Audio(label="Generated Vinyl Sound"),
     title="Vinyl Sound Generator",
-    description="Generate a synthetic vinyl sound using pink noise, rumble, hiss, and pops. Adjust the noise ratio and bandpass frequencies to modify the sound.",
+    description="Generate a synthetic vinyl sound using pink noise, rumble, hiss, and pops. Adjust the noise ratio, bandpass frequencies, duration, and pop rate to modify the sound.",
     examples=[
-        [0.7, 300, 5000],
-        [0.5, 500, 4000],
-        [0.8, 200, 6000]
+        [0.001, 300, 5000, 10, 10],
+        [0.002, 500, 4000, 15, 20],
+        [0.005, 200, 6000, 20, 30]
     ]
 )
 
-iface.launch()
+iface.launch()