|
import numpy as np |
|
import soundfile as sf |
|
from scipy import signal |
|
import gradio as gr |
|
import shutil |
|
import tempfile |
|
from IPython.display import Audio, display |
|
from typing import Tuple |
|
|
|
def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate, rpm, rumble_level, hiss_level): |
|
|
|
sample_rate = 44100 |
|
num_samples = int(duration * sample_rate) |
|
|
|
|
|
wow_freq = rpm / 60 * 0.1 |
|
flutter_freq = np.random.uniform(1, 10) |
|
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) |
|
|
|
|
|
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) |
|
|
|
|
|
nyquist_rate = 0.5 * sample_rate |
|
low = lowcut / nyquist_rate |
|
high = highcut / nyquist_rate |
|
sos = signal.butter(1, [low, high], btype='band', output='sos') |
|
pink_noise = signal.sosfilt(sos, pink_noise) |
|
|
|
|
|
rumble_noise = np.random.normal(0, 1, num_samples) |
|
sos = signal.butter(2, 0.005, btype='low', output='sos') |
|
rumble_noise = signal.sosfilt(sos, rumble_noise) * rumble_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_level |
|
|
|
|
|
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) |
|
pop_data[pop_times] = pop_amplitudes |
|
|
|
|
|
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) |
|
|
|
|
|
vinyl_sound = noise_ratio * (pink_noise + rumble_noise + hiss_noise) + (1 - noise_ratio) * pop_data |
|
|
|
|
|
vinyl_sound = vinyl_sound * (1 + wow_flutter) |
|
|
|
|
|
vinyl_sound /= np.max(np.abs(vinyl_sound)) |
|
|
|
return vinyl_sound.astype(np.float32), sample_rate |
|
|
|
|
|
def convert_to_wav(data, sample_rate): |
|
|
|
data /= np.max(np.abs(data)) |
|
|
|
|
|
temp_file = tempfile.mktemp(suffix=".wav") |
|
sf.write(temp_file, data, sample_rate) |
|
|
|
return temp_file |
|
|
|
|
|
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 |
|
|
|
|
|
iface = gr.Interface( |
|
fn=play_and_download_sound, |
|
inputs=[ |
|
gr.inputs.Slider(minimum=0, maximum=0.005, default=0.0005, step=0.0001, 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=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.Slider(minimum=0, maximum=10, default=0.1, step=0.01, label="Rumble Level"), |
|
gr.inputs.Slider(minimum=0, maximum=10, default=0.05, step=0.01, label="Hiss Level") |
|
], |
|
outputs=[ |
|
gr.outputs.Audio(type="numpy", label="Vinyl Sound"), |
|
gr.outputs.HTML("<button onclick='downloadSound()'>Download Vinyl Sound</button>" |
|
"<script>function downloadSound(){" |
|
"var link = document.createElement('a');" |
|
"link.href = '/download';" |
|
"link.download = 'vinyl_sound.wav';" |
|
"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, RPM, rumble level, and hiss level to modify the sound.", |
|
examples=[ |
|
[0.0005, 300, 5000, 30, 10, 33.33, 0.1, 0.05], |
|
[0.001, 500, 4000, 30, 50, 45.0, 0.15, 0.08], |
|
[0.0008, 400, 6000, 20, 15, 78.0, 0.2, 0.1] |
|
] |
|
) |
|
|
|
iface.launch() |