Update app.py

app.py

@@ -2,7 +2,9 @@ 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
 
 def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     # Parameters
@@ -59,17 +61,18 @@ def convert_to_wav(data, sample_rate):
 
     return temp_file
 
-def vinyl_sound_app(noise_ratio, lowcut, highcut, duration, pop_rate):
+def play_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
-    return data
+    display(Audio(data, rate=sample_rate))
 
-def play_sound(data):
-    data /= np.max(np.abs(data))
-    audio_html = f'<audio controls><source src="data:audio/wav;base64,{sf.write(data, "temp.wav", sample_rate)}" type="audio/wav"></audio>'
-    return audio_html
+def download_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
+    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut,    duration, pop_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,
+    fn=None,
     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)"),
@@ -78,9 +81,8 @@ iface = gr.Interface(
         gr.inputs.Slider(minimum=1, maximum=100, default=10, step=1, label="Pop Rate (pops per second)")
     ],
     outputs=[
-        gr.outputs.HTML(label="Generated Vinyl Sound", html=True),
-        gr.outputs.Button(label="Download Vinyl Sound"),
-        gr.outputs.HTML(label="Play Vinyl Sound")
+        gr.outputs.Button(label="Play Vinyl Sound", type="button", callback=play_sound),
+        gr.outputs.Button(label="Download Vinyl Sound", type="button", callback=download_sound)
     ],
     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.",
@@ -91,11 +93,4 @@ iface = gr.Interface(
     ]
 )
 
-iface.layout = "vertical"
-
-def download_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
-    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
-    temp_file = convert_to_wav(data, sample_rate)
-    return temp_file
-
 iface.launch()