Update app.py

app.py

@@ -59,32 +59,58 @@ def convert_to_wav(data, sample_rate):
 
     return temp_file
 
-def download_wav():
-    data, sample_rate = generate_vinyl_sound(0.001, 300, 5000, 10, 10)
+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
+
 iface = gr.Interface(
-    generate_vinyl_sound,
-    [
-        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)")
-    ],
-    [
-        gr.outputs.Audio(label="Generated Vinyl Sound", type="numpy"),
-        gr.outputs.Button(label="Download Vinyl Sound", type="auto")
-    ],
+    fn=vinyl_sound_app,
+    inputs=["slider", "slider", "slider", "slider", "slider"],
+    outputs="audio",
     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,
+    live=False,
+    template=html_template
 )
 
-iface.launch(download_button=download_wav)
+iface.launch(inbrowser=True)