Update app.py

app.py

@@ -12,7 +12,7 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
     sample_rate = 44100  # sample rate in Hz
     num_samples = int(duration * sample_rate)
 
-    # Calculate wow and flutter based on RPM
+    # Calculate wow and flutter based on RPM (more realistic variations)
     wow_freq = rpm / 60 * 0.1  # Example: 0.1 Hz variation for 33 1/3 RPM
     flutter_freq = np.random.uniform(1, 10)  # Flutter frequency in Hz
     wow_flutter = 0.001 * np.sin(2 * np.pi * wow_freq * np.arange(num_samples) / sample_rate) + \
@@ -21,7 +21,7 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
     # Generate pink noise (optimized using faster method)
     pink_noise = np.random.normal(0, 1, num_samples)
     b, a = signal.butter(2, [0.002, 0.4], btype='band')
-    pink_noise = signal.filtfilt(b, a, pink_noise)  # Use filtfilt for zero-phase filtering
+    pink_noise = signal.filtfilt(b, a, pink_noise)
 
     # Apply band-pass filter to pink noise (optimized using sosfilt)
     nyquist_rate = 0.5 * sample_rate
@@ -30,10 +30,10 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
     sos = signal.butter(1, [low, high], btype='band', output='sos')
     pink_noise = signal.sosfilt(sos, pink_noise)
 
-    # Generate low-frequency rumble (optimized using sosfilt)
+    # Generate low-frequency rumble (optimized using sosfilt, more pronounced)
     rumble_noise = np.random.normal(0, 1, num_samples)
-    sos = signal.butter(2, 0.002, btype='low', output='sos')
-    rumble_noise = signal.sosfilt(sos, rumble_noise)
+    sos = signal.butter(2, 0.005, btype='low', output='sos')  # Lower cutoff for more rumble
+    rumble_noise = signal.sosfilt(sos, rumble_noise) * 0.1  # Adjust amplitude
 
     # Generate high-frequency hiss (optimized using sosfilt)
     hiss_noise = np.random.normal(0, 1, num_samples)
@@ -44,16 +44,16 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
     num_pops = int(duration * pop_rate)
     pop_times = np.random.randint(0, num_samples, num_pops)
     pop_data = np.zeros(num_samples)
-    pop_amplitudes = np.random.uniform(0.05, 0.2, num_pops)  # Vary pop amplitudes
-    pop_data[pop_times] = pop_amplitudes  # Apply random loudness
+    pop_amplitudes = np.random.uniform(0.05, 0.2, num_pops)
+    pop_data[pop_times] = pop_amplitudes
 
     # Apply pop filter (optimized by applying to the entire signal at once)
-    pop_filter_freq = np.random.uniform(0.05, 0.2)  # Vary filter frequency
+    pop_filter_freq = np.random.uniform(0.05, 0.2)
     b, a = signal.butter(4, pop_filter_freq)
     pop_data = signal.lfilter(b, a, pop_data)
 
     # Combine the noises and pops
-    vinyl_sound = noise_ratio * (pink_noise + 0.05 * rumble_noise + 0.05 * hiss_noise) + (1 - noise_ratio) * pop_data
+    vinyl_sound = noise_ratio * (pink_noise + 0.1 * rumble_noise + 0.05 * hiss_noise) + (1 - noise_ratio) * pop_data
 
     # Apply wow and flutter
     vinyl_sound = vinyl_sound * (1 + wow_flutter)
@@ -89,7 +89,7 @@ iface = gr.Interface(
         gr.inputs.Slider(minimum=20, maximum=20000, default=5000, step=10, label="Highcut Frequency (Hz)"),
         gr.inputs.Slider(minimum=1, maximum=600, default=30, step=1, label="Duration (seconds)"),
         gr.inputs.Slider(minimum=1, maximum=180, default=10, step=1, label="Pop Rate (pops per second)"),
-        gr.inputs.Dropdown([33.33, 45], default=33.33, label="RPM")
+        gr.inputs.Number(default=33.33, label="RPM") 
     ],
     outputs=[
         gr.outputs.Audio(type="numpy", label="Vinyl Sound"),
@@ -104,7 +104,8 @@ iface = gr.Interface(
     description="Generate a synthetic vinyl sound using pink noise, rumble, hiss, and pops. Adjust the noise ratio, bandpass frequencies, duration, pop rate, and RPM to modify the sound.",
     examples=[
         [0.0005, 300, 5000, 30, 10, 33.33],  # Example for 33 1/3 RPM
-        [0.001, 500, 4000, 30, 50, 45]      # Example for 45 RPM
+        [0.001, 500, 4000, 30, 50, 45.0],      # Example for 45 RPM
+        [0.0008, 400, 6000, 20, 15, 78.0]      # Example for 78 RPM
     ]
 )