Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -359,9 +359,9 @@ def apply_mask(d):
|
|
359 |
global masks
|
360 |
|
361 |
masks[frame_selected] = d["background"]
|
362 |
-
return masks[frame_selected]
|
363 |
|
364 |
-
def draw_mask(
|
365 |
global params
|
366 |
global frame_selected
|
367 |
global masks
|
@@ -490,14 +490,14 @@ with gr.Blocks(css=css) as demo:
|
|
490 |
output_switch = gr.Checkbox(label="Show depths")
|
491 |
output_switch.input(fn=switch_rows, inputs=[output_switch], outputs=[output_frame])
|
492 |
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")
|
493 |
-
|
|
|
494 |
with gr.Accordion(label="Edge", open=False):
|
495 |
-
gradients = gr.Files(visible=False, value=['./gradient_large.png', './gradient.png', './gradient_small.png'])
|
496 |
levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)
|
497 |
contour = gr.Slider(label="Contour width", value=3, maximum=15, minimum=1, step=2)
|
498 |
ksize = gr.Slider(label="Average", value=7, maximum=16, minimum=0, step=1)
|
499 |
mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
|
500 |
-
mouse.input(fn=draw_mask, show_progress="minimal", inputs=[
|
501 |
selector = gr.HTML(value="""
|
502 |
<a href='#' id='selector' onclick='this.style.fontWeight=\"bold\";
|
503 |
document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].oncontextmenu = function(e){e.preventDefault();}
|
|
|
359 |
global masks
|
360 |
|
361 |
masks[frame_selected] = d["background"]
|
362 |
+
return masks[frame_selected], masks[frame_selected]
|
363 |
|
364 |
+
def draw_mask(l, c, k, v, d, evt: gr.EventData):
|
365 |
global params
|
366 |
global frame_selected
|
367 |
global masks
|
|
|
490 |
output_switch = gr.Checkbox(label="Show depths")
|
491 |
output_switch.input(fn=switch_rows, inputs=[output_switch], outputs=[output_frame])
|
492 |
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")
|
493 |
+
depth_file = gr.File(label="Edited depth")
|
494 |
+
output_mask.apply(fn=apply_mask, inputs=[output_mask], outputs=[output_mask, depth_file])
|
495 |
with gr.Accordion(label="Edge", open=False):
|
|
|
496 |
levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)
|
497 |
contour = gr.Slider(label="Contour width", value=3, maximum=15, minimum=1, step=2)
|
498 |
ksize = gr.Slider(label="Average", value=7, maximum=16, minimum=0, step=1)
|
499 |
mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
|
500 |
+
mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, contour, ksize, mouse, output_mask], outputs=[output_mask])
|
501 |
selector = gr.HTML(value="""
|
502 |
<a href='#' id='selector' onclick='this.style.fontWeight=\"bold\";
|
503 |
document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].oncontextmenu = function(e){e.preventDefault();}
|