Update app.py

app.py

@@ -361,12 +361,9 @@ def draw_mask(f, n, c, v, d, evt: gr.EventData):
     else:
         gradient = cv2.imread(f[2]).astype(np.uint8)
 
-    if not d["layers"][1]:
+    if len(d["layers"]) == 1:
         d["layers"].append(np.zeros_like(d["layers"][0]))
 
-    d["layers"][0] = np.zeros_like(d["layers"][0])
-    print(d["layers"][0].shape[0])
-
     #for i in range(n):
     #    im[(im >= i*255/n) & (im < (i+1)*255/n)] = i*255/(n-1)
     indices = np.arange(0,256)   # List of all colors 
@@ -397,8 +394,7 @@ def draw_mask(f, n, c, v, d, evt: gr.EventData):
     abs_grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
 
     mask = mask + cv2.inRange(abs_grad, 1, 255)
-    d["layers"][0] = bg.copy()
-    d["layers"][0][mask>0] = (255,255,255,255)
+    bg[mask>0] = (255,255,255,255)
 
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
@@ -409,6 +405,8 @@ def draw_mask(f, n, c, v, d, evt: gr.EventData):
     fill = cv2.floodFill(mask, mask_, (x, y), 1, 0, 0, (
         4 | cv2.FLOODFILL_FIXED_RANGE #| cv2.FLOODFILL_MASK_ONLY | 255 << 8
     ))[1] # 255 << 8 tells to fill with the value 255)
+    
+    d["layers"][0] = bg
     d["layers"][1][fill>0] = (0,0,0,255)
     
     return gr.ImageEditor(value=d)