import random
import time
from colorama import init, Fore, Back, Style

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

def draw_tree(height=10, delay=0.5):
    """
    Анимация новогодней ёлки с эффектами.
    :param height: Высота ёлки.
    :param delay: Задержка между кадрами (сек).
    """
    snow = [" ", ".", "*"]               # Символы для снега
    ornaments = [
        Fore.GREEN + "*",                # Зелёная звезда
        Fore.RED + "o",                  # Красный шарик
        Fore.YELLOW + "x",               # Жёлтая гирлянда
        Fore.CYAN + "+",                 # Голубая снежинка
        Fore.WHITE + Style.BRIGHT + ".", # Белая точка
    ]
    try:
        while True:
            # Очистка консоли
            print("\033c", end="")
            
            # Генерация снега
            snow_background = "".join(random.choice(snow) for _ in range(height * 2 + 5))
            print(snow_background)

            # Заголовок
            print(Fore.CYAN + Style.BRIGHT + "       Новогодняя ёлка\n")

            # Построение ёлки
            for i in range(1, height + 1):
                # Уровень ёлки
                level = " " * (height - i) + "*" * (2 * i - 1)
                decorated_level = "".join(
                    random.choice(ornaments) if char == "*" else char for char in level
                )
                print(decorated_level)

            # Ствол
            print(" " * (height - 1) + Fore.MAGENTA + Style.BRIGHT + "|||")

            # Поздравление
            print("\n" + Fore.YELLOW + Style.BRIGHT + "С Новым годом и счастья в 2025 году :D")

            time.sleep(delay)  # Пауза для анимации
    except KeyboardInterrupt:
        print("\nАнимация остановлена. С наступающим Новым годом!")


if __name__ == "__main__":
    try:
        # Пользовательские настройки
        user_height = int(input("Введите высоту ёлки (по умолчанию 10): ") or 10)
        user_delay = float(input("Введите скорость анимации (сек, по умолчанию 0.5): ") or 0.5)
        draw_tree(height=user_height, delay=user_delay)
    except ValueError:
        print("Пожалуйста, вводите корректные значения.")