Update app.py

app.py

@@ -8,7 +8,7 @@ import matplotlib.pyplot as plt
 import librosa
 import librosa.display
 import os
-import moviepy.editor as mp
+import cv2
 
 # Function to generate frequency visualization frames from audio
 def generate_frequency_visualization(audio_path, fps, num_bars):
@@ -41,10 +41,10 @@ def generate_frequency_visualization(audio_path, fps, num_bars):
 
         # Generate and save each frame
         for i, heights in enumerate(bar_heights):
-            plt.figure(figsize=(10, 6))
-            plt.bar(range(num_bars), heights, color=plt.cm.viridis(np.linspace(0, 1, num_bars)))
-            plt.ylim(0, np.max(S))
-            plt.axis('off')
+            fig, ax = plt.subplots(figsize=(10, 6))
+            ax.bar(range(num_bars), heights, color=plt.cm.viridis(np.linspace(0, 1, num_bars)))
+            ax.set_ylim(0, np.max(S))
+            ax.axis('off')
             plt.savefig(f'frames/frame_{i:04d}.png', bbox_inches='tight', pad_inches=0)
             plt.close()
 
@@ -64,17 +64,27 @@ def create_video_from_frames(frames_directory, audio_path, fps, duration):
         if not frame_files:
             raise ValueError("No frames found to create the video.")
 
-        # Create a video from the frames
-        clip = mp.ImageSequenceClip(frame_files, fps=fps)
+        # Get video dimensions from the first frame
+        first_frame = cv2.imread(frame_files[0])
+        height, width, _ = first_frame.shape
+
+        # Initialize video writer
         video_path = 'output_video.mp4'
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        video_writer = cv2.VideoWriter(video_path, fourcc, fps, (width, height))
+
+        # Write frames to video
+        for frame_file in frame_files:
+            frame = cv2.imread(frame_file)
+            video_writer.write(frame)
 
-        # Add the audio to the video
-        audio_clip = mp.AudioFileClip(audio_path)
-        final_clip = clip.set_audio(audio_clip.subclip(0, duration))
-        final_clip.write_videofile(video_path, codec='libx264', audio_codec='aac')
+        video_writer.release()
+
+        # Merge audio with video using ffmpeg
+        os.system(f"ffmpeg -i {video_path} -i {audio_path} -c:v copy -c:a aac -strict experimental output_with_audio.mp4 -y")
 
         print(f"Video created with {len(frame_files)} frames.")
-        return video_path
+        return 'output_with_audio.mp4'
     except Exception as e:
         print(f"Error creating video from frames: {e}")
         return None
@@ -113,4 +123,5 @@ if __name__ == "__main__":
 # - librosa
 # - numpy
 # - matplotlib
-# - moviepy
+# - opencv-python
+# - ffmpeg (installed separately)