Update app.py

app.py

@@ -34,15 +34,18 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     b, a = signal.butter(2, 0.4, btype='high')
     hiss_noise = signal.lfilter(b, a, hiss_noise)
 
-    # Generate pops
+    # Generate pops with varying loudness and frequency
     num_pops = int(duration * pop_rate)
     pop_times = np.random.randint(0, num_samples, num_pops)
     pop_data = np.zeros(num_samples)
-    pop_data[pop_times] = np.random.normal(0, 1, num_pops) * 0.2  # random loudness
-
-    # Create a simple low-pass filter to make the pops sound more like clicks
-    b, a = signal.butter(4, 0.1)
-    pop_data = signal.lfilter(b, a, pop_data)
+    pop_amplitudes = np.random.uniform(0.05, 0.2, num_pops)  # Vary pop amplitudes
+    pop_data[pop_times] = pop_amplitudes  # Apply random loudness
+    
+    # Create a low-pass filter for clicks, with varying filter frequencies for more realistic pops
+    for i in range(num_pops):
+        pop_filter_freq = np.random.uniform(0.05, 0.2)  # Vary filter frequency
+        b, a = signal.butter(4, pop_filter_freq)
+        pop_data[pop_times[i]] = signal.lfilter(b, a, pop_data[pop_times[i]])
 
     # Combine the noises and pops
     vinyl_sound = noise_ratio * (pink_noise + 0.05 * rumble_noise + 0.05 * hiss_noise) + (1 - noise_ratio) * pop_data
@@ -94,4 +97,4 @@ iface = gr.Interface(
     ]
 )
 
-iface.launch()
+iface.launch()