Update split_video.py

split_video.py

@@ -1,25 +1,26 @@
 # split_video.py
-from moviepy.editor import VideoFileClip
+import subprocess
+import math
 import os
 
 def split_video(input_path, output_dir, segment_duration=60):
     if not os.path.exists(input_path):
         raise ValueError("輸入影片不存在")
-
-    video = VideoFileClip(input_path)
-    total_duration = video.duration
-    num_segments = int(total_duration / segment_duration) + (1 if total_duration % segment_duration > 0 else 0)
+    
     os.makedirs(output_dir, exist_ok=True)
-    outputs = []
+    cmd_duration = ['ffprobe', '-v', 'quiet', '-show_entries', 'format=duration', '-of', 'csv=p=0', input_path]
+    result = subprocess.run(cmd_duration, capture_output=True, text=True)
+    total_duration = float(result.stdout.strip())
 
+    num_segments = math.ceil(total_duration / segment_duration)
+    outputs = []
     for i in range(num_segments):
-        start = i * segment_duration
-        end = min((i + 1) * segment_duration, total_duration)
-        subclip = video.subclip(start, end)
+        start_time = i * segment_duration
         output_file = os.path.join(output_dir, f"segment_{i+1}.mp4")
-        subclip.write_videofile(output_file, codec='libx264', audio_codec='aac', verbose=False, logger=None)
+        cmd_split = [
+            'ffmpeg', '-i', input_path, '-ss', str(start_time), '-t', str(segment_duration),
+            '-c', 'copy', '-avoid_negative_ts', 'make_zero', output_file, '-y'
+        ]
+        subprocess.run(cmd_split, check=True)
         outputs.append(output_file)
-        subclip.close()
-
-    video.close()
     return outputs