Update app.py

app.py

@@ -2,9 +2,7 @@ import numpy as np
 import soundfile as sf
 from scipy import signal
 import gradio as gr
-import shutil
 import tempfile
-import simpleaudio as sa
 
 def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     # Parameters
@@ -61,73 +59,37 @@ def convert_to_wav(data, sample_rate):
 
     return temp_file
 
-def play_wav(noise_ratio, lowcut, highcut, duration, pop_rate):
-    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
-    play_obj = sa.play_buffer((data * 32767).astype(np.int16), 1, 2, sample_rate)
-    play_obj.wait_done()
-
-def download_wav(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"
-
-# HTML template for the custom UI
-html_template = """
-<div style="padding: 20px;">
-    <h2>Vinyl Sound Generator</h2>
-    <p>Adjust the parameters to generate a synthetic vinyl sound:</p>
-    <form method="POST" action="/" target="_blank">
-        <label for="noise_ratio">Noise Ratio:</label>
-        <input type="range" id="noise_ratio" name="noise_ratio" min="0" max="1" step="0.001" value="0.001">
-        <br><br>
-        <label for="lowcut">Lowcut Frequency (Hz):</label>
-        <input type="range" id="lowcut" name="lowcut" min="20" max="20000" step="10" value="300">
-        <br><br>
-        <label for="highcut">Highcut Frequency (Hz):</label>
-        <input type="range" id="highcut" name="highcut" min="20" max="20000" step="10" value="5000">
-        <br><br>
-        <label for="duration">Duration (seconds):</label>
-        <input type="range" id="duration" name="duration" min="1" max="60" step="1" value="10">
-        <br><br>
-        <label for="pop_rate">Pop Rate (pops per second):</label>
-        <input type="range" id="pop_rate" name="pop_rate" min="1" max="100" step="1" value="10">
-        <br><br>
-        <input type="submit" value="Generate Vinyl Sound">
-    </form>
-    <br>
-    <button onclick="playSound()">Play Vinyl Sound</button>
-</div>
-
-<script>
-    function playSound() {
-        var audio = new Audio("generated_vinyl_sound.wav");
-        audio.play();
-    }
-</script>
-"""
-
 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)
-    shutil.move(temp_file, "generated_vinyl_sound.wav")
-    return data
+    return temp_file
+
+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>'
+    return audio_html
 
 iface = gr.Interface(
     fn=vinyl_sound_app,
-    inputs=["slider", "slider", "slider", "slider", "slider"],
-    outputs=[gr.outputs.Audio(label="Generated Vinyl Sound"), gr.outputs.Button(label="Download Vinyl Sound")],
+    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=[
+        gr.outputs.HTML(label="Generated Vinyl Sound", type="html", html=True),
+        gr.outputs.File(label="Download Vinyl Sound", type="auto", convert=convert_to_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]
-    ],
-    allow_flagging=False,
-    layout="horizontal",
-    capture_session=True,
-    template=html_template
+    ]
 )
 
-iface.launch(inbrowser=True)
+iface.launch()