Update app.py

app.py

@@ -604,7 +604,7 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
 
-    sel = cv2.floodFill(d["layers"][0], None, (x, y), 1, 2, 2, (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
+    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]