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

def generate_logo(text, font_size, background_color, text_color):
    # Создаем изображение с белым фоном
    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()
    
    # Рассчитываем положение текста
    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="Текст логотипа"),
        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="Создайте простой логотип с текстом."
)

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