Update app.py

app.py

@@ -26,11 +26,26 @@ def remove_watermark_area(original_image, text_mask_path):
     cleaned_image = cv2.GaussianBlur(inpainted_image, (3, 3), 0)
 
     return cleaned_image
+from PIL import Image
 
 def remove_watermark(image_path,saved_path):
-    # Load the original image
-    image = cv2.imread(image_path)
-
+    if isinstance(image_path, str) and os.path.isfile(image_path):
+        # Load the image using OpenCV
+        image = cv2.imread(image_path) 
+    elif isinstance(image_path, np.ndarray):
+        # Directly use OpenCV image (NumPy array)
+        image = image_path
+        if len(image_path.shape) == 3 and image_path.shape[2] == 3:
+            # Assuming it's in RGB format; convert to BGR
+            image = cv2.cvtColor(image_path, cv2.COLOR_RGB2BGR)
+        else:
+            # Otherwise, assume it's already in BGR format
+            image = image_path
+    else:
+        raise TypeError("Invalid image_path format")
+    print(type(image))
+    cv2.imwrite("test.jpg",image)
+    image=cv2.resize(image,(1280,1280))
     # Define the area of the watermark (adjust this based on the watermark size)
     height, width, _ = image.shape
     watermark_width = 185  # Adjust based on your watermark size
@@ -65,6 +80,12 @@ def random_image_name():
     return str(uuid.uuid4())[:8]
 
     
+def process_file(pil_image):
+    saved_path = f"./temp/{random_image_name()}.jpg"
+    remove_watermark(pil_image, saved_path)
+    return saved_path, saved_path
+
+
 def process_files(image_files):
     image_list = []
     if len(image_files) == 1:
@@ -83,14 +104,29 @@ def process_files(image_files):
         zip_path = make_zip(image_list)
         return zip_path,None
 
+
 if not os.path.exists("./temp"):
     os.mkdir("./temp")
 
-demo = gr.Interface(
+
+meta_examples = ["./images/1.jpg", "./images/2.jpg", "./images/3.jpg", "./images/4.jpg", "./images/5.jpg", "./images/6.jpg"]
+
+gradio_input=[gr.Image()]
+gradio_Output=[gr.File(),gr.Image()]
+gradio_interface = gr.Interface(fn=process_file, inputs=gradio_input,outputs=gradio_Output ,
+                              title="Meta Watermark Remover",
+                              examples=meta_examples)
+# gradio_interface.launch(debug=True)
+
+
+
+gradio_multiple_images = gr.Interface(
     process_files,
     [gr.File(type='filepath', file_count='multiple')],
     [gr.File(),gr.Image()],
     cache_examples=True
 )
 
+demo = gr.TabbedInterface([gradio_interface, gradio_multiple_images], ["Meta Watermark Remover","Bluk Meta Watermark Remover"],title="Meta Watermark Remover")
 demo.launch(debug=True)
+