from PIL import Image, ImageDraw, ImageFont
import gradio as gr
import time

def generate_logo(text, font_size, background_color, text_color, industry, description):
    # Задержка 12 секунд
    time.sleep(12)
    
    # Создаем изображение с заданным фоном
    width, height = 400, 200
    image = Image.new('RGB', (width, height), background_color)
    
    # Создаем объект для рисования
    draw = ImageDraw.Draw(image)
    
    # Загружаем шрифт (если шрифт не найден, используется стандартный)
    try:
        font = ImageFont.truetype("arial.ttf", font_size)
    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)
    
    # Добавляем информацию об отрасли и описании (опционально)
    info = f"Отрасль: {industry}\nОписание: {description}"
    draw.text((10, height - 40), info, font=ImageFont.load_default(), fill="gray")
    
    # Сохраняем изображение
    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"),
        gr.Textbox(label="Отрасль", placeholder="Введите отрасль (например, IT, рестораны, мода)"),
        gr.Textbox(label="О чем ваш логотип?", placeholder="Краткое описание логотипа")
    ],
    outputs=gr.Image(label="Сгенерированный логотип"),
    title="Генератор логотипов",
    description="Создайте простой логотип с текстом. Логотип будет сгенерирован через 12 секунд."
)

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