Update app.py

app.py

@@ -400,6 +400,9 @@ def blur_image(image, depth, blur_data):
         blur_frame[blur_mask>0] = blur[blur_mask>0]
         j = j + 1
 
+    white = cv2.inRange(blur_frame, np.array([255,255,255]), np.array([255,255,255]))
+    blur_frame[white>0] = (254,254,254)
+    
     return blur_frame
 
 def loadfile(f):
@@ -617,12 +620,12 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
 
     mask = cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY)
     mask[mask==0] = 128
-    print(mask[mask>0]-128)
+    print(mask[mask>128]-128)
     mask, bgdModel, fgdModel = cv2.grabCut(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2RGB), mask-128, None,None,None,15, cv2.GC_INIT_WITH_MASK)
-    mask = np.where((mask==2)|(mask==0),1,0).astype('uint8')
+    mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
 
     frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
-    frame[:, :, 3] = mask * 255
+    frame[mask>0] = (255,255,255)
     cv2.imwrite(frames[frame_selected], frame)
     
     return gr.ImageEditor(value=d)