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 None  # Если изображение не найдено, возвращаем None

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(value=load_image, label="Pipeline изображение")

    # Устанавливаем действия для кнопки
    submit_btn.click(simple_function, inputs=name_input, outputs=greeting_output)

    demo.launch()