Update app.py

app.py

@@ -4,61 +4,70 @@ from scipy.io.wavfile import read, write
 import tempfile
 
 def dynamic_vibration_simulation(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 to range [-1, 1]
-    data = data / np.max(np.abs(data))
-    
-    # Parameters for vibration simulation
-    segment_duration = 0.1  # Analyze 0.1-second segments
-    segment_length = int(sample_rate * segment_duration)
+    try:
+        # Read the uploaded WAV file
+        sample_rate, data = read(file)
+        
+        # Debug: Print sample rate and shape
+        print(f"Sample Rate: {sample_rate}, Data Shape: {data.shape}")
 
-    # Initialize the low-frequency waveform
-    low_freq_wave = []
+        # Convert to mono if stereo
+        if len(data.shape) > 1:
+            data = data.mean(axis=1)
+        
+        # Normalize the data to range [-1, 1]
+        data = data / np.max(np.abs(data))
+        
+        # Parameters for vibration simulation
+        segment_duration = 0.1  # Analyze 0.1-second segments
+        segment_length = int(sample_rate * segment_duration)
 
-    for i in range(0, len(data), segment_length):
-        segment = data[i:i + segment_length]
+        # Initialize the low-frequency waveform
+        low_freq_wave = []
 
-        if len(segment) < segment_length:
-            break  # Skip incomplete segment at the end
+        for i in range(0, len(data), segment_length):
+            segment = data[i:i + segment_length]
 
-        # Calculate amplitude (RMS value) of the segment
-        rms_amplitude = np.sqrt(np.mean(segment**2))
+            if len(segment) < segment_length:
+                break  # Skip incomplete segment at the end
 
-        # Map amplitude to vibration parameters
-        if rms_amplitude > 0.5:
-            freq = 50  # Strong vibration frequency (50 Hz)
-            duration = segment_duration
-        elif rms_amplitude > 0.2:
-            freq = 30  # Medium vibration frequency (30 Hz)
-            duration = segment_duration
-        else:
-            freq = 0  # Silence (no vibration)
-            duration = segment_duration
+            # Calculate amplitude (RMS value) of the segment
+            rms_amplitude = np.sqrt(np.mean(segment**2))
+            print(f"RMS Amplitude for segment {i // segment_length}: {rms_amplitude}")
 
-        # Generate waveform for this segment
-        t = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
-        if freq > 0:
-            wave = 0.5 * np.sin(2 * np.pi * freq * t)
-        else:
-            wave = np.zeros_like(t)  # Silence
+            # Map amplitude to vibration parameters
+            if rms_amplitude > 0.5:
+                freq = 50  # Strong vibration frequency (50 Hz)
+                duration = segment_duration
+            elif rms_amplitude > 0.2:
+                freq = 30  # Medium vibration frequency (30 Hz)
+                duration = segment_duration
+            else:
+                freq = 0  # Silence (no vibration)
+                duration = segment_duration
+
+            # Generate waveform for this segment
+            t = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
+            if freq > 0:
+                wave = 0.5 * np.sin(2 * np.pi * freq * t)
+            else:
+                wave = np.zeros_like(t)  # Silence
+            
+            # Append the generated wave
+            low_freq_wave.extend(wave)
+        
+        # Convert the waveform to 16-bit PCM
+        low_freq_wave = (np.array(low_freq_wave) * 32767).astype(np.int16)
         
-        # Append the generated wave
-        low_freq_wave.extend(wave)
-    
-    # Convert the waveform to 16-bit PCM
-    low_freq_wave = (np.array(low_freq_wave) * 32767).astype(np.int16)
-    
-    # Save the generated waveform to a temporary file
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
-    write(temp_file.name, sample_rate, low_freq_wave)
+        # Save the generated waveform to a temporary file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
+        write(temp_file.name, sample_rate, low_freq_wave)
 
-    return temp_file.name
+        print("Output WAV file generated successfully.")
+        return temp_file.name
+    except Exception as e:
+        print(f"Error: {e}")
+        return "Error in processing the uploaded WAV file."
 
 # Gradio Interface
 interface = gr.Interface(