Update app.py

app.py

@@ -1,57 +1,43 @@
 import gradio as gr
 import numpy as np
-from scipy.io.wavfile import read, write
+from scipy.io.wavfile import write
 import tempfile
 
-def simulate_vibration(file):
-    # Read the uploaded WAV file
-    sample_rate, data = read(file)
-    
-    # Convert to mono if stereo
-    if len(data.shape) > 1:
-        data = data.mean(axis=1)
-    
-    # Normalize the data
-    data = data / np.max(np.abs(data))
-    
-    # Extract vibration pattern based on amplitude
-    vibration_pattern = []
-    for amp in data[::sample_rate // 10]:  # Take 10 samples per second
-        if abs(amp) > 0.5:
-            vibration_pattern.append(40)  # Strong vibration frequency (40 Hz)
-        elif abs(amp) > 0.2:
-            vibration_pattern.append(20)  # Weak vibration frequency (20 Hz)
-        else:
-            vibration_pattern.append(0)   # No vibration
-
-    # Generate low-frequency vibration-like sound
-    duration_per_burst = 0.1  # 0.1 seconds per vibration burst
-    t = np.linspace(0, duration_per_burst, int(sample_rate * duration_per_burst), endpoint=False)
-    low_freq_wave = []
-
-    for freq in vibration_pattern:
-        if freq > 0:
-            wave = 0.5 * np.sin(2 * np.pi * freq * t)  # Generate sine wave
-        else:
-            wave = np.zeros_like(t)  # Silence
-        low_freq_wave.extend(wave)
-    
-    # Convert to 16-bit PCM
-    low_freq_wave = (np.array(low_freq_wave) * 32767).astype(np.int16)
-    
-    # Save the new simulated vibration audio to a temporary file
+def simulate_amber_alert_vibration(file):
+    # Sampling parameters
+    sample_rate = 44100  # Standard audio sample rate (44.1 kHz)
+    burst_frequency = 50  # Low frequency (50 Hz) for vibration-like sound
+    burst_duration = 0.5  # Duration of each burst in seconds
+    silence_duration = 0.5  # Silence between bursts in seconds
+    num_bursts = 5  # Number of vibration bursts
+
+    # Generate a single burst of low-frequency sound
+    t_burst = np.linspace(0, burst_duration, int(sample_rate * burst_duration), endpoint=False)
+    burst_wave = 0.5 * np.sin(2 * np.pi * burst_frequency * t_burst)  # 50 Hz sine wave
+
+    # Generate silence
+    t_silence = np.linspace(0, silence_duration, int(sample_rate * silence_duration), endpoint=False)
+    silence_wave = np.zeros_like(t_silence)
+
+    # Combine bursts and silence to simulate vibration pattern
+    vibration_wave = np.concatenate([np.concatenate([burst_wave, silence_wave]) for _ in range(num_bursts)])
+
+    # Normalize to 16-bit PCM format
+    vibration_wave = (vibration_wave * 32767).astype(np.int16)
+
+    # Save to a temporary WAV file
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
-    write(temp_file.name, sample_rate, low_freq_wave)
+    write(temp_file.name, sample_rate, vibration_wave)
 
     return temp_file.name
 
 # Gradio Interface
 interface = gr.Interface(
-    fn=simulate_vibration,
-    inputs=gr.Audio(label="Upload a WAV file"),
-    outputs=gr.Audio(label="Simulated Vibration WAV"),
-    title="Simulate Vibrations Using Low-Frequency Audio",
-    description="Upload a WAV file, and a low-frequency audio file will be generated to simulate vibrations."
+    fn=simulate_amber_alert_vibration,
+    inputs=gr.Audio(label="Upload a WAV file (optional)", type="filepath"),
+    outputs=gr.Audio(label="Simulated Amber Alert Vibration"),
+    title="Amber Alert Vibration Simulation",
+    description="Simulate an Amber Alert-style vibration using low-frequency audio. Play the generated audio to feel the effect."
 )
 
 interface.launch()