app.py

import gradio as gr
from funcs import process_image

def process_image_gradio(image_path, watermark_text, font_path, font_size, opacity, color, right_margin, bottom_margin):
 # Вызываем функцию process_image из funcs.py
 output_paths = process_image(image_path, "output", watermark_text, font_path.name, font_size, opacity, color, right_margin, bottom_margin)

 # Cover obj to json str
 generation_params_str = f"Watermark Text: {watermark_text}\nFont Size: {font_size}\nOpacity: {opacity}\nColor: {color}\nRight Margin: {right_margin}\nBottom Margin: {bottom_margin}"

 # Возвращаем путь к изображению с водяным знаком
 return output_paths[0], output_paths[1], generation_params_str

# Определяем интерфейс с использованием Gradio
with gr.Blocks() as demo:
 with gr.Row():
 with gr.Column():
 # Компоненты для настроек
 image = gr.Image(label="Upload Image", type="filepath")
 font_path = gr.File(label="Font Path, .ttf only", file_types=[".ttf"])
 watermark_text = gr.Textbox(label="Watermark Text")
 font_size = gr.Slider(minimum=6, maximum=100, step=1, label="Font Size", value=30)
 opacity = gr.Slider(minimum=0, maximum=1, step=0.01, label="Opacity", value=0.5)
 color = gr.ColorPicker(label="Color",value="#000000")
 right_margin = gr.Slider(minimum=0, maximum=1, step=0.01, label="Right Margin", value=0.05)
 bottom_margin = gr.Slider(minimum=0, maximum=1, step=0.01, label="Bottom Margin", value=0.02)

 with gr.Column():
 # Компонент для отображения результата
 process_button = gr.Button("Process Image")
 generation_params = gr.Textbox(label="Generation Parameters",interactive=False,show_copy_button=True)
 output_image = gr.Image(label="Output Image",interactive=False)
 output_mini = gr.Image(label="Output Image (Mini)",interactive=False)
 
 

 # Обработчик нажатия кнопки "Process Image"
 process_button.click(process_image_gradio, inputs=[image, watermark_text, font_path, font_size, opacity, color, right_margin, bottom_margin], outputs=[output_image,output_mini,generation_params])

# Запускаем интерфейс
demo.queue()
demo.launch()