28. Sept. 2024, 18:25

app.py

@@ -315,12 +315,11 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
 
                 ## 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():
                         with gr.Column():
                             invert_toggle_info = ["Schwarzer Hintergrund, Weiße Linien", "Weißer Hintergrund, Schwarze Linien"]
                             preprocessor_name = gr.Radio(label="Preprocessor", show_label=False, choices=["Lineart", "Lineart Coarse", "Lineart Anime"], type="value", value="Lineart")
+
                             invert = gr.Checkbox(label="Farbe invertieren?", info=invert_toggle_info[0], value=True, elem_classes="toggle-btn")
                             resolution = gr.Slider(label="Auflösung", minimum=256, maximum=1024, value=512, step=64)
                             invert.change(lambda x: {"info": invert_toggle_info[0] if x else invert_toggle_info[1], "__type__": "update"}, inputs=invert, outputs=invert)
@@ -335,10 +334,10 @@ 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)
 
-                run_btn_lineart = gr.Button("Run", visible=False)
-                run_btn_canny = gr.Button("Run", visible=False)
+                run_btn_lineart = gr.Button("Los", variant="primary", visible=True)
+                run_btn_canny = gr.Button("Los", variant="primary", visible=False)
 
-                # all_run_btns = [run_btn_lineart, run_btn_canny]
+                all_run_btns = [run_btn_lineart, run_btn_canny]
 
                 def set_inputs(tab):
                     tab = "tab_lineart" if tab is None else tab
@@ -349,6 +348,7 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                     return {"visible": lineart_btn_visible, "__type__": "update"}, {"visible": canny_btn_visible, "__type__": "update"}
 
                 tab_lineart.select(fn=lambda: set_inputs("tab_lineart"), inputs=None, outputs=[run_btn_lineart, run_btn_canny])
+                tab_canny.select(fn=lambda: set_inputs("tab_canny"), inputs=None, outputs=[run_btn_lineart, run_btn_canny])
 
             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")