Update app.py

app.py

@@ -41,6 +41,7 @@ gradient = None
 params = { "fnum":0 }
 pcolors = []
 frame_selected = 0
+comb_frames = []
 frames = []
 backups = []
 depths = []
@@ -150,7 +151,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
         #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 comb_frames
         count = 0
         n = 0
         depth_frames = []
@@ -455,6 +457,9 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
     depth = cv2.imread(depths[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
     depth[mask>0] = (0,0,0,0)
     cv2.imwrite(depths[frame_selected], depth)
+
+    comb_frame = np.concatenate(frame, depth, axis=0)
+    cv2.imwrite(comb_frames[frame_selected], comb_frame)
     
     return gr.ImageEditor(value=d)
 
@@ -474,8 +479,8 @@ def get_mean_and_std(x):
 
 def color_transfer(base):
     global frames
-	targets = []
-
+    targets = []
+    
 	for n in range(len(frames)):
       targets.append(frames[base])
       if n != base: