renderer/video.py

import moviepy.editor as mp

# import moviepy.video.fx.all as vfx
import os


class Video:
    def __init__(self, frame_path: str, fps: float = 20.0, res="high"):
        frame_path = str(frame_path)
        self.fps = fps

        self._conf = {
            "codec": "libx264",
            "fps": self.fps,
            "audio_codec": "aac",
            "temp_audiofile": "temp-audio.m4a",
            "remove_temp": True,
        }

        if res == "low":
            bitrate = "500k"
        else:
            bitrate = "5000k"

        self._conf = {"bitrate": bitrate, "fps": self.fps}

        # Load video
        # video = mp.VideoFileClip(video1_path, audio=False)
        # Load with frames
        frames = [os.path.join(frame_path, x) for x in sorted(os.listdir(frame_path))]
        video = mp.ImageSequenceClip(frames, fps=fps)
        self.video = video
        self.duration = video.duration

    def save(self, out_path):
        out_path = str(out_path)
        self.video.subclip(0, self.duration).write_videofile(
            out_path, verbose=False, logger=None, **self._conf
        )