Update app.py

app.py

@@ -61,13 +61,11 @@ def convert_to_wav(data, sample_rate):
 
 def vinyl_sound_app(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
+    return data
 
-def play_sound(file_path):
-    with open(file_path, "rb") as f:
-        audio_data = f.read()
-    audio_html = f'<audio controls src="data:audio/wav;base64,{audio_data.encode("base64")}"></audio>'
+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
 
 iface = gr.Interface(
@@ -80,8 +78,9 @@ 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", type="html", html=True),
-        gr.outputs.File(label="Download Vinyl Sound", type="auto", convert=convert_to_wav)
+        gr.outputs.HTML(label="Generated Vinyl Sound", html=True),
+        gr.outputs.Button(label="Download Vinyl Sound"),
+        gr.outputs.HTML(label="Play Vinyl 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.",
@@ -92,4 +91,11 @@ 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()