You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
1.0 KiB
28 lines
1.0 KiB
from __future__ import annotations
|
|
|
|
from typing import Tuple
|
|
|
|
from moviepy.video.VideoClip import VideoClip, TextClip
|
|
from moviepy.video.io.VideoFileClip import VideoFileClip
|
|
from moviepy.video.compositing.CompositeVideoClip import CompositeVideoClip
|
|
|
|
|
|
class Video:
|
|
def __init__(self, video: VideoClip | VideoFileClip, *args, **kwargs):
|
|
self.video: VideoClip = video
|
|
self.fps = self.video.fps
|
|
self.duration = self.video.duration
|
|
|
|
@staticmethod
|
|
def _create_watermark(text, fontsize, opacity=0.5):
|
|
txt_clip = TextClip(text, fontsize=fontsize, color='black').set_opacity(opacity)
|
|
return txt_clip
|
|
|
|
def add_watermark(self, text, opacity=0.5, position: Tuple = (0.95, 0.95), fontsize=15):
|
|
txt_clip = self._create_watermark(text, opacity=opacity, fontsize=fontsize)
|
|
txt_clip = txt_clip.set_pos(position).set_duration(10)
|
|
|
|
# Overlay the text clip on the first video clip
|
|
self.video = CompositeVideoClip([self.video, txt_clip])
|
|
return self.video
|