Update app.py

app.py

@@ -169,6 +169,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         cv2.destroyAllWindows()
 
         global frame_selected
+        global masks
+        masks = []
         return final_vid, final_zip, orig_frames, depth_frames, depth_frames[frame_selected] #output_path
 
 def depth_edges_mask(depth):
@@ -304,12 +306,18 @@ def loadurl(url):
 
 def select_frame(v, f, evt: gr.SelectData):
     global frame_selected
+    global masks
+    j = 0
+    while j < len(f):
+        if not masks[j]:
+            masks[j] = None
+        j = j + 1
     masks[frame_selected] = v
     
     if evt.index != frame_selected:
         frame_selected = evt.index
 
-    if masks[frame_selected]:
+    if masks[frame_selected] != None:
         v = masks[frame_selected]
     else:
         v = f[frame_selected][0]