from moviepy.editor import VideoFileClip, CompositeVideoClip, ImageClip
from PIL import Image, ImageDraw, ImageFont
import numpy as np


def add_watermark(input_video, output_video, watermark_text, font_size=24, position=("right", "bottom"), opacity=128):
    # Загружаем видео
    video = VideoFileClip(input_video)

    # Создаём изображение с текстом водяного знака
    font_path = "arial.ttf"  # Указываем путь к файлу шрифта
    font = ImageFont.truetype(font_path, font_size)

    # Получаем размеры текста
    text_bbox = font.getbbox(watermark_text)
    text_width, text_height = text_bbox[2], text_bbox[3]

    # Создаём прозрачное изображение для водяного знака
    watermark_image = Image.new("RGBA", (text_width, text_height), (0, 0, 0, 0))
    draw = ImageDraw.Draw(watermark_image)
    draw.text((0, 0), watermark_text, font=font, fill=(255, 255, 255, opacity))

    # Преобразуем Pillow-изображение в NumPy-массив
    watermark_array = np.array(watermark_image)

    # Конвертируем NumPy-массив в ImageClip
    watermark_clip = ImageClip(watermark_array).set_duration(video.duration)

    # Определяем позицию водяного знака
    if position == ("right", "bottom"):
        position = (video.size[0] - text_width - 10, video.size[1] - text_height - 10)
    elif position == ("left", "top"):
        position = (10, 10)
    elif position == ("right", "top"):
        position = (video.size[0] - text_width - 10, 10)
    elif position == ("left", "bottom"):
        position = (10, video.size[1] - text_height - 10)

    watermark_clip = watermark_clip.set_position(position)

    # Создаем итоговое видео
    final_video = CompositeVideoClip([video, watermark_clip])
    final_video.write_videofile(output_video, codec="libx264", audio_codec="aac")


# Пример использования
add_watermark(
    input_video="input.mp4",
    output_video="output_with_watermark.mp4",
    watermark_text="My Watermark",
    font_size=36,
    position=("right", "bottom"),
    opacity=200
)