Update app.py

app.py

@@ -560,7 +560,7 @@ def reset_mask(d):
     backup = cv2.imread(backups[frame_selected]).astype(np.uint8)
     cv2.imwrite(frames[frame_selected], backup)
 
-    d["layers"][0] = (0,0,0,0)
+    d["layers"][0] = np.zeros(d["layers"][0].shape[:2],np.uint8)
 
     return gr.ImageEditor(value=d)
 
@@ -579,13 +579,12 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
     pts = pts.reshape((-1,1,2))
 
     if len(edge) == 0 or params["fnum"] != frame_selected:
-      if len(edge) > 0:
+      if params["fnum"] != frame_selected:
           d["background"] = cv2.imread(depths[frame_selected]).astype(np.uint8)
+          params["fnum"] = frame_selected
 
       bg = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY)
       bg[bg==255] = 0
-
-      params["fnum"] = frame_selected
         
       edge = bg.copy()
     else: