Update app.py

app.py

@@ -565,16 +565,9 @@ with gr.Blocks(css=css, js=js) 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=('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")
-            with gr.Accordion(label="Edge", open=False):
-              levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)
-              tolerance = gr.Slider(label="Tolerance", value=1, maximum=7, minimum=0, step=1)
-              bsize = gr.Slider(label="Border size", value=15, maximum=256, minimum=1, step=2)
-              mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
-              mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, tolerance, mouse, output_mask], outputs=[output_mask])
-              output_mask.apply(fn=apply_mask, inputs=[output_mask, bsize], outputs=[output_mask, depth_file])
             selector = gr.HTML(value="""
-            <a href='#' id='selector' onclick='this.style.fontWeight=\"bold\";
+            <a href='#' id='selector' onclick='if (this.style.fontWeight!=\"bold\") {
+            this.style.fontWeight=\"bold\";
             document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].oncontextmenu = function(e){e.preventDefault();}
             document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].ondrag = function(e){e.preventDefault();}
             
@@ -600,11 +593,24 @@ with gr.Blocks(css=css, js=js) as demo:
               
               document.getElementById(\"mouse\").getElementsByTagName(\"textarea\")[0].style.borderColor = \"#ffffff\";
             
+            }
+            } else {
+              this.style.fontWeight=\"normal\";
+              document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].onclick = null;
+              
             }' title='Select point' style='text-decoration:none;color:white;'>⊹ Select point</a> <a href='#' id='clear_select' onclick='
               
               document.getElementById(\"mouse\").getElementsByTagName(\"textarea\")[0].value = \"[]\";
             
             ' title='Clear selection' style='text-decoration:none;color:white;'>✕ Clear selection</a>""")
+            depth_file = gr.File(label="Edited depth")
+            with gr.Accordion(label="Edge", open=False):
+              levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)
+              tolerance = gr.Slider(label="Tolerance", value=1, maximum=7, minimum=0, step=1)
+              bsize = gr.Slider(label="Border size", value=15, maximum=256, minimum=1, step=2)
+              mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
+              mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, tolerance, mouse, output_mask], outputs=[output_mask])
+              output_mask.apply(fn=apply_mask, inputs=[output_mask, bsize], outputs=[output_mask, depth_file])
             submit = gr.Button("Submit")
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')