Update app.py

app.py

@@ -359,7 +359,8 @@ def apply_mask(d):
     global masks
     
     masks[frame_selected] = d["background"]
-    return masks[frame_selected], masks[frame_selected]
+    cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])
+    return masks[frame_selected], f"f{frame_selected}_dmap.png"
 
 def draw_mask(l, c, k, v, d, evt: gr.EventData):
     global params
@@ -489,7 +490,7 @@ with gr.Blocks(css=css) as demo:
             output_frame = gr.Gallery(label="Frames", preview=True, columns=8192, type="numpy")
             output_switch = gr.Checkbox(label="Show depths")
             output_switch.input(fn=switch_rows, inputs=[output_switch], outputs=[output_frame])
-            output_mask = gr.ImageEditor(layers=False, sources=None, type="numpy", interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(default_size=0, colors=['black', '#505050', '#a0a0a0', 'white']), elem_id="image_edit")
+            output_mask = gr.ImageEditor(layers=False, sources=('upload', 'clipboard'), show_download_button=True, type="numpy", interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(default_size=0, colors=['black', '#505050', '#a0a0a0', 'white']), elem_id="image_edit")
             depth_file = gr.File(label="Edited depth")
             output_mask.apply(fn=apply_mask, inputs=[output_mask], outputs=[output_mask, depth_file])
             with gr.Accordion(label="Edge", open=False):