Update app.py

app.py

@@ -7,32 +7,38 @@ import tempfile
 from IPython.display import Audio, display
 from typing import Tuple
 
-def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
+def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
     # Parameters
     sample_rate = 44100  # sample rate in Hz
     num_samples = int(duration * sample_rate)
 
-    # Generate pink noise
+    # Calculate wow and flutter based on RPM
+    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) + \
+                  0.0005 * np.sin(2 * np.pi * flutter_freq * np.arange(num_samples) / sample_rate)
+
+    # 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.lfilter(b, a, pink_noise)
+    pink_noise = signal.filtfilt(b, a, pink_noise)  # Use filtfilt for zero-phase filtering
 
-    # Apply band-pass filter to pink noise
+    # Apply band-pass filter to pink noise (optimized using sosfilt)
     nyquist_rate = 0.5 * sample_rate
     low = lowcut / nyquist_rate
     high = highcut / nyquist_rate
-    b, a = signal.butter(1, [low, high], btype='band')
-    pink_noise = signal.lfilter(b, a, pink_noise)
+    sos = signal.butter(1, [low, high], btype='band', output='sos')
+    pink_noise = signal.sosfilt(sos, pink_noise)
 
-    # Generate low-frequency rumble
+    # Generate low-frequency rumble (optimized using sosfilt)
     rumble_noise = np.random.normal(0, 1, num_samples)
-    b, a = signal.butter(2, 0.002, btype='low')
-    rumble_noise = signal.lfilter(b, a, rumble_noise)
+    sos = signal.butter(2, 0.002, btype='low', output='sos')
+    rumble_noise = signal.sosfilt(sos, rumble_noise)
 
-    # Generate high-frequency hiss
+    # Generate high-frequency hiss (optimized using sosfilt)
     hiss_noise = np.random.normal(0, 1, num_samples)
-    b, a = signal.butter(2, 0.4, btype='high')
-    hiss_noise = signal.lfilter(b, a, hiss_noise)
+    sos = signal.butter(2, 0.4, btype='high', output='sos')
+    hiss_noise = signal.sosfilt(sos, hiss_noise)
 
     # Generate pops with varying loudness and frequency
     num_pops = int(duration * pop_rate)
@@ -41,22 +47,23 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     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 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)
-        
-        # Apply the filter to the entire pop_data array
-        pop_data = signal.lfilter(b, a, pop_data)
+    # 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
+    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
 
+    # Apply wow and flutter
+    vinyl_sound = vinyl_sound * (1 + wow_flutter)
+
     # Normalize to between -1 and 1
     vinyl_sound /= np.max(np.abs(vinyl_sound))
 
     return vinyl_sound.astype(np.float32), sample_rate
 
+
 def convert_to_wav(data, sample_rate):
     # Normalize to between -1 and 1
     data /= np.max(np.abs(data))
@@ -67,11 +74,13 @@ def convert_to_wav(data, sample_rate):
 
     return temp_file
 
-def play_and_download_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
-    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
+
+def play_and_download_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
+    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm)
     temp_file = convert_to_wav(data, sample_rate)
     return temp_file, temp_file
 
+
 iface = gr.Interface(
     fn=play_and_download_sound,
     inputs=[
@@ -79,7 +88,8 @@ iface = gr.Interface(
         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=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.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")
     ],
     outputs=[
         gr.outputs.Audio(type="numpy", label="Vinyl Sound"),
@@ -91,11 +101,10 @@ iface = gr.Interface(
                         "link.click();}</script>")
     ],
     title="Vinyl Sound Generator",
-    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.",
+    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],
-        [0.001, 500, 4000, 30, 50],
-        [0.002, 200, 6000, 30, 100]
+        [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
     ]
 )