Update app.py

app.py

@@ -482,17 +482,21 @@ def apply_mask(d, b):
     d["background"][mask==255] = (0,0,0,0)
     mask = cv2.inRange(d["background"], np.array([0,0,0,0]), np.array([0,0,0,0]))
 
+    ksize = int(np.sqrt(np.sum(mask) / 255))
+    if ksize/2 - int(ksize/2) == 0:
+        ksize = ksize + 1
+    dmask = cv2.GaussianBlur(dmask, (ksize,ksize), 0)
+
     for k, mk in enumerate(masks):
         if k != frame_selected and k < len(depths):
             masks[k] = cv2.imread(depths[k]).astype(np.uint8)
             masks[k][dmask<255] = (0,0,0)
-
-    ksize = int(np.sqrt(np.sum(mask) / 255))
-    if ksize/2 - int(ksize/2) == 0:
-        ksize = ksize + 1
+            cv2.imwrite(depths[k], masks[k])
+            frame = cv2.imread(frames[k], cv2.IMREAD_UNCHANGED).astype(np.uint8)
+            frame[:, :, 3] = dmask
+            cv2.imwrite(frames[k], frame)
         
     frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
-    dmask = cv2.GaussianBlur(dmask, (ksize,ksize), 0)
     frame[:, :, 3] = 255 - dmask
     cv2.imwrite(frames[frame_selected], frame)