Update app.py

app.py

@@ -4,6 +4,7 @@ from scipy import signal
 import gradio as gr
 import shutil
 import tempfile
+import IPython.display as ipd
 
 def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     # Parameters
@@ -66,24 +67,31 @@ def download_wav(noise_ratio, lowcut, highcut, duration, pop_rate):
     shutil.move(temp_file, "download.wav")
     return "download.wav"
 
+def play_wav(noise_ratio, lowcut, highcut, duration, pop_rate):
+    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
+    ipd.display(ipd.Audio(data, rate=sample_rate))
+    return None
+
 iface = gr.Interface(
-    fn=download_wav,
+    fn=None,  # We don't need a main function
     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]
-]
+        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=[
+        gr.outputs.File(label="Download Vinyl Sound", type="auto", convert=download_wav),
+        gr.outputs.Button(label="Play Vinyl Sound", type="auto", callback=play_wav)
+    ],
+    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)
-