Update app.py

app.py

@@ -21,6 +21,7 @@ from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 from moviepy.editor import *
 
 frame_selected = 0
+masks = []
 locations = []
 mesh = []
 
@@ -303,11 +304,17 @@ def loadurl(url):
 
 def select_frame(v, f, evt: gr.SelectData):
     global frame_selected
+    masks[frame_selected] = v
+    
     if evt.index != frame_selected:
         frame_selected = evt.index
+
+    if masks[frame_selected]:
+        v = masks[frame_selected]
+    else:
+        v = f[frame_selected][0]
     #print(v)
-    print(f[frame_selected][0])
-    return gr.Gallery(selected_index=evt.index, preview=True), f[frame_selected][0], frame_selected
+    return gr.Gallery(selected_index=evt.index, preview=True), v, frame_selected
 
 
 css = """