import cv2
from scenedetect import detect, ContentDetector
from moviepy.editor import VideoFileClip

import sys
# 检测视频中的场景
scene_list = detect(sys.argv[1], ContentDetector())

frame_list = []
    
for i, scene in enumerate(scene_list):
    if i==0:
        print(scene[0].get_frames(),scene[1].get_frames()+2)
        frame_list.append([scene[0].get_frames(),scene[1].get_frames()+2])
    elif i==len(scene_list)-1:
        print(scene[0].get_frames()+2,scene[1].get_frames())
        frame_list.append([scene[0].get_frames()+2,scene[1].get_frames()])
    else:
        print(scene[0].get_frames()+2,scene[1].get_frames()+2)
        frame_list.append([scene[0].get_frames()+2,scene[1].get_frames()+2])


# 打开原始视频
video_clip = VideoFileClip(sys.argv[1])

# 遍历场景并保存为新的 MP4 文件
for i, (start_frame, end_frame) in enumerate(scene_list):
    # 将帧数转换为秒
    start_time = start_frame.get_seconds()  # 使用 get_seconds() 转换为秒
    end_time = end_frame.get_seconds()      # 使用 get_seconds() 转换为秒
    
    # 切割视频
    scene_clip = video_clip.subclip(start_time, end_time)
    
    # 设置输出文件名
    output_filename = f"oringal_{i+1}.mp4"
    
    # 保存场景为 MP4 文件
    scene_clip.write_videofile(output_filename, codec="libx264", fps=video_clip.fps)

    print(f"Scene {i+1} has been saved as {output_filename}")

# 关闭视频文件
video_clip.close()

# 打开原始视频
video_clip = VideoFileClip(sys.argv[1])
fps =video_clip.fps
# 遍历场景并保存为新的 MP4 文件
for index,item in enumerate(frame_list):
    # 将帧数转换为秒
    start_time = item[0]/fps  
    end_time = item[1]/fps  
    
    # 切割视频
    scene_clip = video_clip.subclip(start_time, end_time)
    
    # 设置输出文件名
    output_filename = f"modified_{index+1}.mp4"
    
    # 保存场景为 MP4 文件
    scene_clip.write_videofile(output_filename, codec="libx264", fps=video_clip.fps)

    print(f"Scene {i+1} has been saved as {output_filename}")

# 关闭视频文件
video_clip.close()