zfff

requirements.txt

@@ -3,4 +3,5 @@ librosa
 numpy
 requests
 stability-sdk
-opencv-python
+Opencv-python-headless
+replicate

video_generator.py

@@ -1,4 +1,6 @@
 from PIL import Image, ImageDraw, ImageFont
+import cv2
+
 
 def generate_lyric_frame(text, image_size):
     # Create an image object
@@ -15,15 +17,46 @@ def generate_lyric_frame(text, image_size):
     draw.text((text_x, text_y), text, font=font, fill='white')
     return image
 
-def generate_video(lyrics_with_timing, image_size=(800, 600), max_frames=48):
-    # Placeholder for configuring the animation
-    # ...
-
+def generate_video(lyrics_with_timing, image_size=(800, 600), frame_rate=30):
     frames = []
     for start_time, end_time, lyric in lyrics_with_timing:
         # Calculate the number of frames for this lyric segment
+        num_frames = int((end_time - start_time) * frame_rate)
+
+        # Create frames with the lyric text
+        for _ in range(num_frames):
+            frame = generate_lyric_frame(lyric, image_size)
+            frames.append(frame)
+
+    # Combine frames into a video (replace with your preferred method)
+    video = combine_frames_into_video(frames, frame_rate)
+    return video
 
 def extract_last_frame_from_video(video):
     # Your code to extract the last frame from the video
     last_frame_image = None  # Replace with actual image object
     return last_frame_image
+
+def combine_frames_into_video(frames, frame_rate):
+    # Get the size of the frames
+    height, width, _ = frames[0].shape
+
+    # Define the codec and create a video writer object
+    fourcc = cv2.VideoWriter_fourcc(*'XVID')
+    out = cv2.VideoWriter('output.avi', fourcc, frame_rate, (width, height))
+
+    # Write each frame to the video file
+    for frame in frames:
+        # Convert PIL image to NumPy array if the frames are PIL images
+        frame_np = np.array(frame)
+        
+        # Convert RGB to BGR as OpenCV uses BGR format
+        frame_bgr = cv2.cvtColor(frame_np, cv2.COLOR_RGB2BGR)
+        
+        # Write the BGR frame to the video
+        out.write(frame_bgr)
+
+    # Release the video writer object
+    out.release()
+
+    return 'output.avi'