Update app.py

app.py

@@ -592,10 +592,6 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
 
       params["fnum"] = frame_selected
       params["l"] = l
-
-      mask = cv2.imread(masks[frame_selected], cv2.IMREAD_GRAYSCALE).astype(np.uint8) + 128
-      mask[mask==128] = 0
-      d["layers"][0] = cv2.cvtColor(mask, cv2.COLOR_GRAY2RGBA)
         
       edge = bg.copy()
     else:
@@ -604,6 +600,10 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
 
+    mask = cv2.imread(masks[frame_selected], cv2.IMREAD_GRAYSCALE).astype(np.uint8) + 128
+    mask[mask==128] = 0
+    d["layers"][0] = cv2.cvtColor(mask, cv2.COLOR_GRAY2RGBA)
+
     sel = cv2.floodFill(mask, None, (x, y), 1, 2, 2, (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
     # 255 << 8 tells to fill with the value 255)
     sel = sel[1:sel.shape[0]-1, 1:sel.shape[1]-1]