Update app.py

app.py

@@ -1,34 +1,28 @@
-# Импортируем необходимые библиотеки
 import gradio as gr
 import cv2
-import numpy as np
+from watermark_removal import remove_watermark
 
-# Определяем функцию, которая удаляет watermark с изображения
-def remove_watermark(image):
-  # Преобразуем изображение в серый цвет
-  gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-  # Применяем медианный фильтр для сглаживания шума
-  gray = cv2.medianBlur(gray, 5)
-  # Находим границы водяного знака с помощью алгоритма Canny
-  edges = cv2.Canny(gray, 50, 150)
-  # Находим контуры водяного знака с помощью алгоритма findContours
-  contours, hierarchy = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-  # Создаем маску для водяного знака с помощью алгоритма drawContours
-  mask = np.zeros_like(gray)
-  cv2.drawContours(mask, contours, -1, 255, -1)
-  # Применяем инпейнтинг для заполнения водяного знака цветом фона с помощью алгоритма inpaint
-  result = cv2.inpaint(image, mask, 3, cv2.INPAINT_TELEA)
-  # Возвращаем результат
-  return result
+# Функция для удаления водяных знаков с изображения
+def remove_watermark(input_image):
+    # Загрузка изображения
+    image = cv2.imdecode(np.frombuffer(input_image.read(), np.uint8), -1)
+    
+    # Удаление водяных знаков с использованием библиотеки image-watermark-removal
+    result_image = remove_watermark(image)
 
-# Создаем интерфейс Gradio с помощью библиотеки gradio
+    # Преобразование изображения обратно в байты
+    _, 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(), # Указываем тип выходного параметра - изображение
-  title="RWOI", # Указываем заголовок приложения
-  description="This app removes watermark from images using OpenCV." # Указываем описание приложения
+    fn=remove_watermark,
+    inputs=gr.Image(),
+    outputs=gr.Image(),
+    live=True,
+    capture_session=True
 )
 
-# Запускаем приложение
-iface.launch()
+iface.launch()