Update app.py

app.py

@@ -2,88 +2,85 @@ 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
+import shutil
 
 def generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
-    # Parameters
-    sample_rate = 44100  # sample rate in Hz
-    num_samples = int(duration * sample_rate)
-
-    # Generate pink noise
-    pink_noise = np.random.normal(0, 1, num_samples)
-    b, a = signal.butter(2, [0.002, 0.4], btype='band')
-    pink_noise = signal.lfilter(b, a, pink_noise)
-
-    # Apply band-pass filter to pink noise
-    nyquist_rate = 0.5 * sample_rate
-    low = lowcut / nyquist_rate
-    high = highcut / nyquist_rate
-    b, a = signal.butter(1, [low, high], btype='band')
-    pink_noise = signal.lfilter(b, a, pink_noise)
-
-    # Generate low-frequency rumble
-    rumble_noise = np.random.normal(0, 1, num_samples)
-    b, a = signal.butter(2, 0.002, btype='low')
-    rumble_noise = signal.lfilter(b, a, rumble_noise)
-
-    # Generate high-frequency hiss
-    hiss_noise = np.random.normal(0, 1, num_samples)
-    b, a = signal.butter(2, 0.4, btype='high')
-    hiss_noise = signal.lfilter(b, a, hiss_noise)
-
-    # Generate pops
-    num_pops = int(duration * pop_rate)
-    pop_times = np.random.randint(0, num_samples, num_pops)
-    pop_data = np.zeros(num_samples)
-    pop_data[pop_times] = np.random.normal(0, 1, num_pops) * 0.2  # random loudness
-
-    # Create a simple low-pass filter to make the pops sound more like clicks
-    b, a = signal.butter(4, 0.1)
-    pop_data = signal.lfilter(b, a, pop_data)
-
-    # Combine the noises and pops
-    vinyl_sound = noise_ratio * (pink_noise + 0.05 * rumble_noise + 0.05 * hiss_noise) + (1 - noise_ratio) * pop_data
-
-    # Normalize to between -1 and 1
-    vinyl_sound /= np.max(np.abs(vinyl_sound))
-
-    return vinyl_sound.astype(np.float32), sample_rate
+    # Code to generate vinyl sound
+    # ...
 
 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
+    # Code to convert audio data to WAV file
+    # ...
 
 def play_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
     data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
-    display(Audio(data, rate=sample_rate))
+    temp_file = convert_to_wav(data, sample_rate)
+    audio = gr.outputs.Audio(temp_file, label="Generated Vinyl Sound")
+    return audio
 
 def download_sound(noise_ratio, lowcut, highcut, duration, pop_rate):
-    data, sample_rate = generate_vinyl_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"
 
+# Custom HTML template for the 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>
+    <audio id="audio" controls style="display: none;"></audio>
+    <button onclick="playAudio()">Play Vinyl Sound</button>
+    <a href="download.wav" download><button>Download Vinyl Sound</button></a>
+    <script>
+        function playAudio() {
+            var audio = document.getElementById("audio");
+            audio.src = "{sound_url}";
+            audio.play();
+        }
+    </script>
+</div>
+"""
+
+def vinyl_sound_app(noise_ratio, lowcut, highcut, duration, pop_rate):
+    
+    # Generate the vinyl sound
+    data, sample_rate = generate_vinyl_sound(noise_ratio, lowcut, highcut, duration, pop_rate)
+
+    # Convert audio data to WAV file
+    temp_file = convert_to_wav(data, sample_rate)
+
+    # Update the HTML template with the sound URL
+    html_template_with_sound_url = html_template.replace("{sound_url}", temp_file)
+
+    # Return the HTML template
+    return html_template_with_sound_url
+
+# Define the Gradio interface
 iface = gr.Interface(
-    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)"),
-        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.Button(label="Play Vinyl Sound", type="button", callback=play_sound),
-        gr.outputs.Button(label="Download Vinyl Sound", type="button", callback=download_sound)
-    ],
+    fn=vinyl_sound_app,
+    inputs=["slider", "slider", "slider", "slider", "slider"],
+    outputs="html",
     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=[
@@ -93,4 +90,5 @@ iface = gr.Interface(
     ]
 )
 
-iface.launch()
+# Launch the interface
+iface.launch(inbrowser=True)