import gradio as gr
import cv2
from watermark_removal import remove_watermark

# Функция для удаления водяных знаков с изображения
def remove_watermark(input_image):
    # Загрузка изображения
    image = cv2.imdecode(np.frombuffer(input_image.read(), np.uint8), -1)
    
    # Удаление водяных знаков с использованием библиотеки image-watermark-removal
    result_image = remove_watermark(image)

    # Преобразование изображения обратно в байты
    _, encoded_image = cv2.imencode(".jpg", result_image)
    result_image = encoded_image.tobytes()

    return result_image

# Создание Gradio приложения
iface = gr.Interface(
    fn=remove_watermark,
    inputs=gr.Image(),
    outputs=gr.Image(),
    live=True,
    capture_session=True
)

iface.launch()