28. Sept. 2024, 17:00

app.py

@@ -337,22 +337,24 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                             high_threshold = gr.Slider(label="high_threshold", minimum=1, maximum=255, value=200, step=1)
                             resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
 
-                def set_inputs(tab_name):
-                    global input_options
-                    if tab_name == "Lineart":
-                        input_options = [preprocessor_name, invert]
-                    return {"preprocessor_name": preprocessor_name, "invert": invert, "resolution": resolution}
+                def set_inputs(tab, all_run_btns):
+                    # global input_options
+
+                    lineart_btn = all_run_btns[0]
+                    canny_btn = all_run_btns[1]
+
+                    lineart_btn_visible = True if tab == "tab_lineart" else False
+                    canny_btn_visible = True if tab == "tab_canny" else False
+                    {"visible": lineart_btn_visible, "__type__": "update"}, {"visible": canny_btn_visible, "__type__": "update"}
+
+                    return all_run_btns
 
                 run_btn_lineart = gr.Button("Run", visible=False)
                 run_btn_canny = gr.Button("Run", visible=False)
 
                 all_run_btns = [run_btn_lineart, run_btn_canny]
 
-                tab_lineart.select(
-                    fn=set_inputs("lineart", all_run_btns),
-                    inputs=None,
-                    outputs=all_run_btns,
-                )
+                tab_lineart.select(set_inputs("tab_lineart"), None, all_run_btns)
 
             with gr.Column(scale=2):
                 gallery = gr.Gallery(label="Generated images", show_label=False, interactive=False, format="png", elem_id="output_gallery", elem_classes="output-gallery", columns=[3], rows=[2], object_fit="contain", height="auto", type="filepath")