Update app.py

app.py

@@ -388,7 +388,8 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
 
       #for i in range(l):
       #    bg[(bg >= i*255/l) & (bg < (i+1)*255/l)] = i*255/(l-1)
-      bg = cv2.convertScaleAbs(palette[bg] + mask).astype(np.uint8) # Converting image back to uint
+      bg = cv2.convertScaleAbs(palette[bg]).astype(np.uint8) # Converting image back to uint
+      bg[mask>0] = 255
 
       grad_x = cv2.convertScaleAbs(cv2.Sobel(bg, ddepth, 1, 0, ksize=3, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT))
       grad_y = cv2.convertScaleAbs(cv2.Sobel(bg, ddepth, 0, 1, ksize=3, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT))
@@ -407,7 +408,7 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
 
       d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_GRAY2RGBA)
     else:
-      mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
+      mask = cv2.inRange(cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY), 255, 255)
 
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]