from fontTools.ttLib import TTFont
from PIL import Image, ImageDraw, ImageFont
import gradio as gr
import time
import os

def check_font(font_path):
    try:
        font = TTFont(font_path)
        return True
    except:
        return False

def generate_logo(text, font_size, background_color, text_color):
    # Задержка 12 секунд
    time.sleep(12)
    
    # Создаем изображение с заданным фоном
    width, height = 400, 200
    image = Image.new('RGB', (width, height), background_color)
    
    # Создаем объект для рисования
    draw = ImageDraw.Draw(image)
    
    # Путь к шрифту
    font_path = "arial.ttf"
    if not os.path.exists(font_path) or not check_font(font_path):
        font_path = None  # Используем стандартный шрифт
    
    # Загружаем шрифт
    try:
        font = ImageFont.truetype(font_path, font_size) if font_path else ImageFont.load_default()
    except IOError:
        font = ImageFont.load_default()
    
    # Рассчитываем положение текста
    try:
        # Для Pillow >= 10.0.0 используем textbbox вместо textsize
        bbox = draw.textbbox((0, 0), text, font=font)
        text_width = bbox[2] - bbox[0]
        text_height = bbox[3] - bbox[1]
    except AttributeError:
        # Для старых версий Pillow
        text_width, text_height = draw.textsize(text, font=font)
    
    x = (width - text_width) / 2
    y = (height - text_height) / 2
    
    # Рисуем текст на изображении
    draw.text((x, y), text, font=font, fill=text_color)
    
    # Сохраняем изображение
    image.save("logo.png")
    
    return "logo.png"

# Создаем интерфейс Gradio
iface = gr.Interface(
    fn=generate_logo,
    inputs=[
        gr.Textbox(label="Текст логотипа", value="My Logo"),
        gr.Slider(minimum=10, maximum=100, value=40, label="Размер шрифта"),
        gr.ColorPicker(label="Цвет фона", value="#FFFFFF"),
        gr.ColorPicker(label="Цвет текста", value="#000000")
    ],
    outputs=gr.Image(label="Сгенерированный логотип"),
    title="Генератор логотипов",
    description="Создайте простой логотип с текстом. Логотип будет сгенерирован через 12 секунд."
)

# Запускаем приложение
iface.launch()