Update app.py

app.py

@@ -1,6 +1,9 @@
 import numpy as np
+import soundfile as sf
 from scipy import signal
 import gradio as gr
+import shutil
+import tempfile
 
 def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     # Parameters
@@ -47,25 +50,40 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
 
     return vinyl_sound.astype(np.float32), sample_rate
 
-def vinyl_sound_app(noise_ratio, lowcut, highcut, duration, pop_rate):
+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
+
+def download_wav(noise_ratio, lowcut, highcut, duration, pop_rate):
     data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
-    return data, sample_rate
+    temp_file = convert_to_wav(data, sample_rate)
+    shutil.move(temp_file, "download.wav")
+    return "download.wav"
 
 iface = gr.Interface(
-    fn=vinyl_sound_app,
-    inputs=["slider", "slider", "slider", "slider", "slider"],
-    outputs=["audio", "number"],
-    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.",
-    examples=[
-        [0.001, 300, 5000, 10, 10],
-        [0.002, 500, 4000, 15, 20],
-        [0.005, 200, 6000, 20, 30]
-    ],
-    allow_flagging=False,
-    layout="horizontal",
-    capture_session=True,
-    live=False
+    fn=download_wav,
+    inputs=[
+    gr.inputs.Slider(minimum=0, maximum=1, default=0.001, step=0.001, 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=60, default=10, step=1, label="Duration (seconds)"),
+    gr.inputs.Slider(minimum=1, maximum=100, default=10, step=1, label="Pop Rate (pops per second)")
+],
+outputs="file",
+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.",
+examples=[
+    [0.001, 300, 5000, 10, 10],
+    [0.002, 500, 4000, 15, 20],
+    [0.005, 200, 6000, 20, 30]
+]
 )
 
 iface.launch(inbrowser=True)
+