import gradio as gr
from PIL import Image

def simple_function(name):
    # Просто возвращаем приветственное сообщение
    return f"Привет, {name}!"

def load_image():
    # Загрузка изображения pipeline.png
    try:
        img = Image.open("pipeline.png")
        return img
    except IOError:
        return "Ошибка: Изображение pipeline.png не найдено."

with gr.Blocks() as demo:
    gr.Markdown("# Генерация приветственного сообщения и отображение изображения")

    with gr.Row():
        with gr.Column(scale=1):
            name_input = gr.Textbox(label="Введите ваше имя")
            greeting_output = gr.Textbox(label="Приветствие")
            submit_btn = gr.Button("Отправить")
        
        with gr.Column(scale=1):
            image_output = gr.Image(label="Pipeline изображение")

    # Устанавливаем действия для кнопки
    submit_btn.click(simple_function, inputs=name_input, outputs=greeting_output)
    image_output.update(load_image())  # Загружаем и отображаем изображение при запуске

    demo.launch()