Update app.py

app.py

@@ -364,16 +364,17 @@ def show_json(txt):
     return data[0]["video"]["path"], data[1]["path"], data[2], data[3]["background"]["path"], data[4], data[5]
 
 
-def select_frame(v, evt: gr.SelectData):
+def select_frame(d, evt: gr.SelectData):
     global dcolor
     global frame_selected
     global masks
     global edge
     
     if evt.index != frame_selected:
-        masks[frame_selected] = v
-        frame_selected = evt.index
         edge = []
+        mask = cv2.imread(depths[frame_selected]).astype(np.uint8)
+        cv2.imwrite(masks[frame_selected], cv2.cvtColor(mask, cv2.COLOR_RGB2GRAY))
+        frame_selected = evt.index
 
     if len(dcolor) == 0:
         bg = [127, 127, 127, 255]