Update app.py

app.py

@@ -1,7 +1,7 @@
 import decord
 import os
 import gradio as gr
-from moviepy.editor import VideoFileClip
+import subprocess
 
 def split_video_into_chunks(video_path, chunk_duration=6, output_dir="video_chunks"):
     # Load the video using Decord
@@ -16,25 +16,23 @@ def split_video_into_chunks(video_path, chunk_duration=6, output_dir="video_chun
     # Create output directory
     os.makedirs(output_dir, exist_ok=True)
 
-    # Split the video into chunks
+    # Split the video into chunks using FFmpeg
     chunk_files = []
     for i in range(num_chunks):
-        start_frame = int(i * chunk_duration * fps)
-        end_frame = int((i + 1) * chunk_duration * fps)
-
-        # Ensure end_frame does not exceed total_frames
-        end_frame = min(end_frame, total_frames)
-
-        # Extract frames for the chunk
-        chunk_frames = vr[start_frame:end_frame]
-
-        # Save the chunk as a new video file
+        start_time = i * chunk_duration
+        end_time = (i + 1) * chunk_duration
         output_file = os.path.join(output_dir, f"chunk_{i+1}.mp4")
-        chunk_frames = [frame.asnumpy() for frame in chunk_frames]  # Convert Decord frames to numpy arrays
 
-        # Use moviepy to save the chunk
-        clip = VideoFileClip(video_path).subclip(i * chunk_duration, (i + 1) * chunk_duration)
-        clip.write_videofile(output_file, codec="libx264")
+        # FFmpeg command to extract the chunk
+        command = [
+            "ffmpeg",
+            "-i", video_path,
+            "-ss", str(start_time),
+            "-t", str(chunk_duration),
+            "-c", "copy",
+            output_file
+        ]
+        subprocess.run(command, check=True)
 
         chunk_files.append(output_file)