Update app.py

app.py

@@ -20,6 +20,7 @@ from depth_anything.dpt import DepthAnything
 from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 from moviepy.editor import *
 
+edge = None
 gradient = None
 params = { "fnum":0, "l":16, "c":3, "k":7 }
 frame_selected = 0
@@ -358,6 +359,7 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
     global frame_selected
     global masks
     global gradient
+    global edge
     
     points = json.loads(v)
     pts = np.array(points, np.int32)
@@ -407,8 +409,9 @@ def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
       params["k"] = k
 
       d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_GRAY2RGBA)
+      edge = mask.copy()
     else:
-      mask = cv2.inRange(cv2.cvtColor(d["layers"][0], cv2.COLOR_RGBA2GRAY), 255, 255)
+      mask = edge.copy()
 
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]