Update app.py

app.py

@@ -403,14 +403,15 @@ def apply_mask(d, b):
     edge = []
     mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
     dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
-    mask = cv2.dilate(mask, dilation)
+    dmask = cv2.dilate(mask, dilation)
+
     d["background"][mask==255] = (255,255,255,255)
     mask = cv2.inRange(d["background"], np.array([255,255,255,255]), np.array([255,255,255,255]))
 
     for k, mk in enumerate(masks):
         if k != frame_selected:
             masks[k] = cv2.imread(depths[k]).astype(np.uint8)
-            masks[k][mask==0] = (255,255,255)
+            masks[k][dmask==0] = (255,255,255)
 
     masks[frame_selected] = d["background"]
     cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])