import random
import time
from colorama import init, Fore

init(autoreset=True)  # Инициализация colorama


def draw_tree(height=10):
    try:
        while True:
            print("\033c", end="")  # Очистка консоли
            print("Новогодняя ёлка\n")
            for i in range(1, height + 1):
                level = " " * (height - i) + "*" * (2 * i - 1)
                decorated_level = "".join(
                    random.choice([
                        Fore.GREEN + "*",  # Зелёная звезда
                        Fore.RED + "o",    # Красный шарик
                        Fore.YELLOW + "x"  # Жёлтая гирлянда
                    ]) if char == "*" else char for char in level)
                print(decorated_level)
            print(" " * (height - 1) + Fore.MAGENTA + "|||")  # Ствол
            time.sleep(0.5)  # Пауза для анимации
    except KeyboardInterrupt:
        print("\nАнимация остановлена. С наступающим Новым годом!")


if __name__ == "__main__":
    draw_tree()

from PIL import Image, ImageDraw, ImageFont


def create_image_with_text(text, output_path="output.png", text_size=36, position="center"):
    # Создаём новое изображение размером 800x400, цвет фона - тёмно-синий
    img = Image.new("RGB", (800, 400), color=(73, 109, 137))
    # Инициализируем объект для рисования
    draw = ImageDraw.Draw(img)

    # Загружаем шрифт Arial с указанным размером текста
    font = ImageFont.truetype("arial.ttf", size=text_size)

    # Вычисляем размер текста с помощью textbbox (границы текста в формате (x_min, y_min, x_max, y_max))
    text_bbox = draw.textbbox((0, 0), text, font=font)
    text_width = text_bbox[2] - text_bbox[0]  # Ширина текста
    text_height = text_bbox[3] - text_bbox[1]  # Высота текста

    # Устанавливаем координаты текста в зависимости от выбранного положения
    if position == "center":  # Центр изображения
        text_x = (img.width - text_width) // 2
        text_y = (img.height - text_height) // 2
    elif position == "top-left":  # Верхний левый угол
        text_x, text_y = 10, 10
    elif position == "top-right":  # Верхний правый угол
        text_x = img.width - text_width - 10
        text_y = 10
    elif position == "bottom-left":  # Нижний левый угол
        text_x = 10
        text_y = img.height - text_height - 10
    elif position == "bottom-right":  # Нижний правый угол
        text_x = img.width - text_width - 10
        text_y = img.height - text_height - 10
    else:
        # Если указано некорректное положение, выбрасываем исключение
        raise ValueError(
            "Недопустимое значение для position. Используйте: center, top-left, top-right, bottom-left, bottom-right."
        )

    # Добавляем текст на изображение по рассчитанным координатам
    draw.text((text_x, text_y), text, fill="white", font=font)
    # Сохраняем изображение в файл
    img.save(output_path)
    # Выводим сообщение о сохранении файла
    print(f"Изображение сохранено как {output_path}")


# Запрашиваем у пользователя текст для изображения
text = input("Введите текст для изображения: ")
# Запрашиваем у пользователя размер текста
text_size = int(input("Введите размер текста: "))
# Предлагаем выбрать расположение текста
print(
    "Выберите расположение текста: center, top-left, top-right, bottom-left, bottom-right"
)
# Получаем выбор пользователя
position = input("Введите расположение текста: ").strip()
# Вызываем функцию с указанными параметрами
create_image_with_text(text, text_size=text_size, position=position)

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
)