Update app.py

app.py

@@ -425,16 +425,15 @@ def draw_mask(l, c, k, v, d, evt: gr.EventData):
     #bg_ = cv2.copyMakeBorder(bg, 1, 1, 1, 1, cv2.BORDER_DEFAULT)
     #bg = np.zeros_like(bg)
     
-    _, bg, mask = cv2.floodFill(bg, None, (x, y), 1, 0, 0 #, (
-        #4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
-    ) # 255 << 8 tells to fill with the value 255)
+    mask = cv2.floodFill(bg, None, (x, y), 1, 0, 0)[2] #, (4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
+    # 255 << 8 tells to fill with the value 255)
     grad_x = cv2.convertScaleAbs(cv2.Sobel(mask, ddepth, 1, 0, ksize=3, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT))
     grad_y = cv2.convertScaleAbs(cv2.Sobel(mask, ddepth, 0, 1, ksize=3, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT))
     
     grad = cv2.addWeighted(grad_x, 0.5, grad_y, 0.5, 0)
     mask[cv2.inRange(grad, 1, 255)>0] = 1
     
-    d["layers"][1] = bg
+    d["layers"][1][mask>0] = (0,0,0,255)
     d["background"][mask>0] = (0,0,0,255)
     
     return gr.ImageEditor(value=d)