from demo import Demo
import numpy as np
from inpaint import Inpaint

inpute = Demo.loadImage("./dog/cow_img.bmp")
maskImage = Demo.loadImage("./dog/cow_mask.bmp")
inpute = np.copy(inpute)

height, width = inpute.shape[:2]
mask = np.zeros((height,width),dtype=bool)
for y in range(height):
    for x in range(width):
        mask[y,x] = maskImage[y,x] == 255

for y in range(height):
    for x in range(width):
        if mask[y,x]:
            inpute[y,x,0] = 255
            inpute[y,x,1] = 0
            inpute[y,x,2] = 0

Demo.setValue(inpute,mask)
Demo.display(Demo.input)

output = Inpaint(Demo.input,Demo.mask,2)
Demo.display(output)