fruitpicker01's picture
Update app.py
919365e verified
raw
history blame
1.64 kB
import gradio as gr
from PIL import Image, ImageDraw, ImageFont
def simple_function(name):
# Создание нового изображения с фоновым цветом
img = Image.new('RGB', (200, 100), color = (73, 109, 137))
draw = ImageDraw.Draw(img)
# Попытка загрузить шрифт arial.ttf, если не найден — использование шрифта по умолчанию
try:
font = ImageFont.truetype("arial.ttf", 20)
except IOError:
font = ImageFont.load_default()
return "Ошибка: Шрифт arial.ttf не найден. Используется шрифт по умолчанию."
# Рисование текста на изображении
draw.text((10, 40), f"Привет, {name}!", font=font, fill="white")
# Сохранение изображения
img.save("output_test.png")
return f"Привет, {name}! Изображение сохранено как output_test.png"
# Создание интерфейса Gradio
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=2):
submit_btn.click(simple_function, inputs=name_input, outputs=greeting_output)
demo.launch()