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.
56 lines
2.0 KiB
56 lines
2.0 KiB
from __future__ import annotations
|
|
|
|
from moviepy.video.VideoClip import VideoClip, ImageClip
|
|
from moviepy.video.io.VideoFileClip import VideoFileClip
|
|
from PIL import Image, ImageDraw, ImageFont, ImageEnhance
|
|
|
|
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, path, fontsize=15, opacity=0.5):
|
|
|
|
width = 500
|
|
height = 200
|
|
|
|
white = (255, 255, 255)
|
|
transparent = (0, 0, 0, 0)
|
|
|
|
font = ImageFont.load_default()
|
|
wm = Image.new('RGBA', (width, height), transparent)
|
|
im = Image.new('RGBA', (width, height), transparent) # Change this line too.
|
|
|
|
draw = ImageDraw.Draw(wm)
|
|
w, h = draw.textsize(text, font)
|
|
draw.text(((width - w) / 2, (height - h) / 2), text, white, font)
|
|
en = ImageEnhance.Brightness(wm) # todo alow it to use the fontsize
|
|
mask = en.enhance(1 - opacity)
|
|
im.paste(wm, (25, 25), mask)
|
|
im.save(path)
|
|
|
|
def add_watermark(self, text, opacity=0.5):
|
|
# add a watermark to the video clip with the given text and opacity without importing a new library
|
|
from moviepy.video.compositing.CompositeVideoClip import CompositeVideoClip
|
|
from moviepy.video.compositing.concatenate import concatenate_videoclips
|
|
path = './assets/temp/png/watermark.png'
|
|
self._create_watermark(text, path, opacity=opacity)
|
|
image_clips = []
|
|
image_clips.insert(
|
|
0,
|
|
ImageClip(path)
|
|
.set_duration(self.video.duration)
|
|
#.resize(width=W - 100)
|
|
)
|
|
image_concat = concatenate_videoclips(image_clips).set_position((0.1, 0.1))
|
|
self.video = CompositeVideoClip([self.video, image_concat])
|
|
return self.video
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': # todo delete
|
|
Video._create_watermark('Background Video by Jason(example)', '../assets/temp/png/watermark.png')
|