Update app.py

app.py

@@ -354,6 +354,13 @@ def switch_rows(v):
         print(frames[0])
         return frames
 
+def apply_mask(d):
+    global frame_selected
+    global masks
+    
+    masks[frame_selected] = d["background"]
+    return masks[frame_selected]
+
 def draw_mask(f, l, c, k, v, d, evt: gr.EventData):
     global params
     global frame_selected
@@ -483,6 +490,7 @@ with gr.Blocks(css=css) as demo:
             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.apply(fn=apply_mask, inputs=[output_mask], outputs=[output_mask])
             with gr.Accordion(label="Edge", open=False):
               gradients = gr.Files(visible=False, value=['./gradient_large.png', './gradient.png', './gradient_small.png'])
               levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)