28. Sept. 2024, 14:22

app.py

@@ -250,6 +250,9 @@ def inpaint(image, invert):
 
 
 # MARK: GRADIO UI
+
+input_options = []
+
 with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, title=title) as demo:
     with gr.Row(elem_classes="row-header"):
         gr.Markdown(
@@ -268,7 +271,7 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                 input_image = gr.Image(label="Input Image", type="numpy", height=512)
 
                 ## TAB LINEART
-                with gr.Tab("Lineart"):
+                with gr.Tab("Lineart") as tab_lineart:
                     with gr.Row():
                         gr.Markdown("<p>Check Invert to use with Mochi Diffusion. Inverted image can also be created here for use with ControlNet Scribble.")
                     with gr.Row():
@@ -280,9 +283,14 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                             resolution = gr.Slider(label="Auflösung", minimum=256, maximum=1024, value=512, step=64)
                             run_btn_lineart = gr.Button("Run")
                             invert.change(lambda x: {"info": invert_toggle_info[0] if x else invert_toggle_info[1], "__type__": "update"}, inputs=checkbox, outputs=checkbox)
-
+                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}
+                tab_lineart.select(fn=set_inputs(), inputs=tab_lineart, outputs=None)
                 ## TAB Canny
-                with gr.Tab("Canny Edge"):
+                with gr.Tab("Canny Edge") as tab_canny:
                     with gr.Row():
                         gr.Markdown("## Canny Edge")
                     with gr.Row():
@@ -305,7 +313,7 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                 )
 
     # MARK: Button Runs
-    run_btn_lineart.click(fn=lineart, inputs=[input_image, resolution, preprocessor_name, invert, gallery], outputs=[gallery])
+    run_btn_lineart.click(fn=lineart, inputs=[input_image, resolution, gallery, preprocessor_name, invert], outputs=[gallery])
     run_btn_canny.click(fn=canny, inputs=[input_image, resolution, low_threshold, high_threshold, gallery], outputs=[gallery])