Spaces:
Sleeping
Sleeping
File size: 1,185 Bytes
bd65e34 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import solara
from moviepy.editor import VideoFileClip, concatenate_videoclips
import torch
CODEC = {"codec": "h264_nvenc"} if torch.cuda.is_available() else {}
@solara.memoize(
key=lambda _, disabled, file_name, cache_key: f"{disabled}_{file_name}{cache_key}"
)
def write_full_video(
start_stop: list[dict[str, str]], disabled: dict, file_name: str, cache_key: str
) -> str:
vid_clip = VideoFileClip(f"downloaded/{file_name}.mp4")
clips = []
for i, tstamp in enumerate(start_stop):
if disabled.get(i):
continue
clips.append(vid_clip.subclip(tstamp["start"], tstamp["end"]))
# Concatenate the video clips with transitions
final_clip = concatenate_videoclips(clips)
# Write the final concatenated movie to a file
file = vid_clip.filename.replace("downloaded", "out")
final_clip.write_videofile(file)
return file
@solara.memoize
def write_video(start: str, stop: str, id: int, file_name: str) -> str:
vid_clip = VideoFileClip(f"downloaded/{file_name}.mp4")
clip = vid_clip.subclip(start, stop)
file = f"tmp/{file_name}_{start}_{stop}_{id}.mp4"
clip.write_videofile(file)
return file
|