Update app.py

app.py

@@ -117,6 +117,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         #out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*"avc1"), frame_rate, (output_width, frame_height))
         #fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         #out = cv2.VideoWriter(output_path, fourcc, frame_rate, (output_width, frame_height))
+        global masks
         count=0
         depth_frames = []
         orig_frames = []
@@ -160,8 +161,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
 
             cv2.imwrite(f"f{count}.jpg", raw_frame)
             orig_frames.append(f"f{count}.jpg")
-            global masks
             masks.append(f"f{count}.jpg")
+            
             cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
             depth_frames.append(f"f{count}_dmap.jpg")
             count += 1
@@ -320,7 +321,8 @@ def select_frame(v, evt: gr.SelectData):
         masks[frame_selected] = v
         frame_selected = evt.index
         v = masks[frame_selected]
-    #print(v)
+    print(evt.index)
+    print(frame_selected)
     return v, frame_selected
 
 def switch_rows(v):