Create waveform_visualizer.py

Author: larsdpeder

waveform_visualizer.py

@@ -0,0 +1,124 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from matplotlib.animation import FuncAnimation
+from scipy.io import wavfile
+from moviepy.editor import VideoFileClip, AudioFileClip, CompositeVideoClip
+from pydub import AudioSegment
+import os
+import tempfile
+
+def convert_to_wav(input_file):
+    audio = AudioSegment.from_file(input_file)
+    temp_wav = tempfile.NamedTemporaryFile(suffix='.wav', delete=False)
+    audio.export(temp_wav.name, format='wav')
+    return temp_wav.name
+
+def create_waveform_video(audio_file, output_file):
+    temp_wav_name = None
+    try:
+        # Convert to WAV if not already
+        if not audio_file.lower().endswith('.wav'):
+            print("Converting audio to WAV format...")
+            temp_wav_name = convert_to_wav(audio_file)
+            audio_file = temp_wav_name
+
+        # Read the WAV file
+        sample_rate, audio_data = wavfile.read(audio_file)
+        
+        # If stereo, take the mean of both channels
+        if len(audio_data.shape) > 1:
+            audio_data = np.mean(audio_data, axis=1)
+        
+        # Normalize audio data
+        audio_data = audio_data / np.max(np.abs(audio_data))
+        
+        # Calculate duration
+        duration = len(audio_data) / sample_rate
+        
+        # Set up the figure and axis
+        fig, ax = plt.subplots(figsize=(16, 9), facecolor='black')
+        ax.set_facecolor('black')
+        ax.set_ylim(-1, 1)
+        ax.set_xlim(0, 1)  # Set x-axis from 0 to 1
+        
+        # Add subtle grid
+        ax.grid(color='dimgray', linestyle=':', linewidth=0.5, alpha=0.5)
+        
+        # Remove axis labels
+        ax.set_xticks([])
+        ax.set_yticks([])
+        
+        line, = ax.plot([], [], color='white', lw=2)
+        
+        # Number of points to display
+        num_points = 1000
+        
+        # Animation function
+        def animate(frame):
+            start = frame * sample_rate // 30
+            end = start + num_points
+            x = np.linspace(0, 1, num_points)
+            y = audio_data[start:end]
+            if len(y) < num_points:
+                y = np.pad(y, (0, num_points - len(y)), 'constant')
+            line.set_data(x, y)
+            return line,
+        
+        # Create the animation
+        anim = FuncAnimation(fig, animate, frames=int(30 * duration), 
+                             interval=1000/30, blit=True)
+        
+        # Save the animation as a temporary file
+        temp_video = tempfile.NamedTemporaryFile(suffix='.mp4', delete=False)
+        anim.save(temp_video.name, fps=30, extra_args=['-vcodec', 'libx264'])
+        
+        # Close the matplotlib figure
+        plt.close(fig)
+        
+        # Load the temporary video file
+        video = VideoFileClip(temp_video.name)
+        
+        # Load the audio file
+        audio = AudioFileClip(audio_file)
+        
+        # Set the audio of the video
+        final_video = video.set_audio(audio)
+        
+        # Write the final video file
+        final_video.write_videofile(output_file, codec='libx264', audio_codec='aac')
+        
+        # Close the clips
+        video.close()
+        audio.close()
+        final_video.close()
+        
+        # Remove the temporary video file
+        os.unlink(temp_video.name)
+
+    except Exception as e:
+        print(f"An error occurred: {str(e)}")
+    finally:
+        # Clean up the temporary WAV file if it was created
+        if temp_wav_name and os.path.exists(temp_wav_name):
+            os.unlink(temp_wav_name)
+
+def main():
+    while True:
+        audio_file = input("Enter the path to your audio file: ").strip()
+        if os.path.isfile(audio_file):
+            break
+        else:
+            print("Invalid file path. Please try again.")
+
+    output_file = input("Enter the name for the output MP4 file (default: output.mp4): ").strip()
+    if not output_file:
+        output_file = "output.mp4"
+    elif not output_file.lower().endswith('.mp4'):
+        output_file += '.mp4'
+
+    print(f"Creating video from {audio_file}...")
+    create_waveform_video(audio_file, output_file)
+    print(f"Video created successfully: {output_file}")
+
+if __name__ == "__main__":
+    main()