Update app.py

app.py

@@ -7,7 +7,7 @@ import tempfile
 from IPython.display import Audio, display
 from typing import Tuple
 
-def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
+def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm, rumble_level, hiss_level):
     # Parameters
     sample_rate = 44100  # sample rate in Hz
     num_samples = int(duration * sample_rate)
@@ -30,15 +30,15 @@ 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, more pronounced)
+    # Generate low-frequency rumble (optimized using sosfilt, controllable level)
     rumble_noise = np.random.normal(0, 1, num_samples)
     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
+    rumble_noise = signal.sosfilt(sos, rumble_noise) * rumble_level  # Adjust amplitude
 
-    # Generate high-frequency hiss (optimized using sosfilt)
+    # Generate high-frequency hiss (optimized using sosfilt, controllable level)
     hiss_noise = np.random.normal(0, 1, num_samples)
     sos = signal.butter(2, 0.4, btype='high', output='sos')
-    hiss_noise = signal.sosfilt(sos, hiss_noise)
+    hiss_noise = signal.sosfilt(sos, hiss_noise) * hiss_level  # Adjust amplitude
 
     # Generate pops with varying loudness and frequency
     num_pops = int(duration * pop_rate)
@@ -53,7 +53,7 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
     pop_data = signal.lfilter(b, a, pop_data)
 
     # Combine the noises and pops
-    vinyl_sound = noise_ratio * (pink_noise + 0.1 * rumble_noise + 0.05 * hiss_noise) + (1 - noise_ratio) * pop_data
+    vinyl_sound = noise_ratio * (pink_noise + rumble_noise + hiss_noise) + (1 - noise_ratio) * pop_data
 
     # Apply wow and flutter
     vinyl_sound = vinyl_sound * (1 + wow_flutter)
@@ -65,18 +65,11 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm):
 
 
 def convert_to_wav(data, sample_rate):
-    # Normalize to between -1 and 1
-    data /= np.max(np.abs(data))
-
-    # Save to a temporary .wav file
-    temp_file = tempfile.mktemp(suffix=".wav")
-    sf.write(temp_file, data, sample_rate)
-
-    return temp_file
+    # ... (This function remains the same) ...
 
 
-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)
+def play_and_download_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm, rumble_level, hiss_level):
+    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm, rumble_level, hiss_level)
     temp_file = convert_to_wav(data, sample_rate)
     return temp_file, temp_file
 
@@ -89,7 +82,9 @@ 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.Number(default=33.33, label="RPM") 
+        gr.inputs.Number(default=33.33, label="RPM"),
+        gr.inputs.Slider(minimum=0, maximum=0.2, default=0.1, step=0.01, label="Rumble Level"),
+        gr.inputs.Slider(minimum=0, maximum=0.1, default=0.05, step=0.01, label="Hiss Level")
     ],
     outputs=[
         gr.outputs.Audio(type="numpy", label="Vinyl Sound"),
@@ -101,11 +96,11 @@ 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, pop rate, and RPM 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, RPM, rumble level, and hiss level 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.0],      # Example for 45 RPM
-        [0.0008, 400, 6000, 20, 15, 78.0]      # Example for 78 RPM
+        [0.0005, 300, 5000, 30, 10, 33.33, 0.1, 0.05],  # Example for 33 1/3 RPM
+        [0.001, 500, 4000, 30, 50, 45.0, 0.15, 0.08],      # Example for 45 RPM
+        [0.0008, 400, 6000, 20, 15, 78.0, 0.2, 0.1]      # Example for 78 RPM
     ]
 )