AI Video & Image Watermark Remover: Build Free LaMa & ProPainter Inpainting Engine

import os, cv2, json, base64, io, email, traceback

import numpy as np

from http.server import HTTPServer, BaseHTTPRequestHandler

from PIL import Image



try:

 import onnxruntime as ort

 HAS_ONNX = True

except ImportError:

 HAS_ONNX = False



SCRATCH_DIR = os.path.dirname(os.path.abspath(__file__))

MODEL_PATH = os.path.join(SCRATCH_DIR, "lama_fp32.onnx")



session = None

if HAS_ONNX and os.path.exists(MODEL_PATH):

 try:

 session = ort.InferenceSession(MODEL_PATH, providers=['CPUExecutionProvider'])

 print("🧠 LaMa AI Engine Loaded!")

 except Exception as e:

 print(f"ONNX Load Error: {e}")



def run_lama_inference(img_bgr, mask_gray):

 orig_h, orig_w, _ = img_bgr.shape

 if session is not None:

 img_rgb = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)

 img_512 = cv2.resize(img_rgb, (512, 512))

 mask_512 = cv2.resize(mask_gray, (512, 512))



 img_tensor = (img_512.astype(np.float32) / 255.0).transpose((2, 0, 1))[None, ...]

 mask_tensor = (mask_512 > 20).astype(np.float32)[None, None, ...]



 outputs = session.run(None, {'image': img_tensor, 'mask': mask_tensor})

 out = outputs[0][0].transpose((1, 2, 0))

 out = np.clip(out, 0, 255).astype(np.uint8)



 out_bgr = cv2.cvtColor(out, cv2.COLOR_RGB2BGR)

 out_orig = cv2.resize(out_bgr, (orig_w, orig_h))

 mask_3d = (mask_gray > 20)[:, :, None]

 return np.where(mask_3d, out_orig, img_bgr)

 return cv2.inpaint(img_bgr, (mask_gray > 20).astype(np.uint8)*255, 7, cv2.INPAINT_TELEA)



class Handler(BaseHTTPRequestHandler):

 def do_POST(self):

 if self.path == '/api/inpaint':

 length = int(self.headers['Content-Length'])

 req = json.loads(self.rfile.read(length).decode('utf-8'))

 img_data = base64.b64decode(req['image'].split(',')[1])

 mask_data = base64.b64decode(req['mask'].split(',')[1])



 img_bgr = cv2.cvtColor(np.array(Image.open(io.BytesIO(img_data)).convert('RGB')), cv2.COLOR_RGB2BGR)

 mask_gray = np.array(Image.open(io.BytesIO(mask_data)).convert('L'))



 clean_bgr = run_lama_inference(img_bgr, mask_gray)

 clean_rgb = cv2.cvtColor(clean_bgr, cv2.COLOR_BGR2RGB)

 

 buf = io.BytesIO()

 Image.fromarray(clean_rgb).save(buf, format="PNG")

 b64_clean = "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode('utf-8')



 self.send_response(200)

 self.send_header('Content-Type', 'application/json')

 self.end_headers()

 self.wfile.write(json.dumps({'status': 'success', 'clean_image': b64_clean}).encode('utf-8'))



def main():

 server = HTTPServer(('0.0.0.0', 8080), Handler)

 print("🚀 ZipLoot Server running on port 8080...")

 server.serve_forever()



if __name__ == '__main__':

 main()

{

 "cleanUrls": true,

 "rewrites": [

 { "source": "/watermark-remover", "destination": "/watermark-remover.html" },

 { "source": "/api/inpaint", "destination": "https://ziploot.app/watermark-removerapi/inpaint" },

 { "source": "/api/process_video", "destination": "https://ziploot.app/watermark-removerapi/process_video" }

 ]

}