Update app.py

app.py

@@ -1,9 +1,6 @@
 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
@@ -50,56 +47,14 @@ def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
 
     return vinyl_sound.astype(np.float32), sample_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)
-    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>
-</div>
-"""
-
 def vinyl_sound_app(noise_ratio, lowcut, highcut, duration, pop_rate):
     data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
-    return data
+    return data, sample_rate
 
 iface = gr.Interface(
     fn=vinyl_sound_app,
     inputs=["slider", "slider", "slider", "slider", "slider"],
-    outputs="audio",
+    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=[
@@ -110,8 +65,7 @@ iface = gr.Interface(
     allow_flagging=False,
     layout="horizontal",
     capture_session=True,
-    live=False,
-    template=html_template
+    live=False
 )
 
 iface.launch(inbrowser=True)