28. Sept. 2024, 16:42

app.py

@@ -31,6 +31,8 @@ title = "ControlNet v1.1 Preprocessors Standalone"
 
 # Canny
 model_canny = None
+
+
 def canny(img, res, l, h):
     img = resize_image(HWC3(img), res)
     global model_canny
@@ -41,8 +43,11 @@ def canny(img, res, l, h):
     result = model_canny(img, l, h)
     return [result]
 
+
 # Hed
 model_hed = None
+
+
 def hed(img, res):
     img = resize_image(HWC3(img), res)
     global model_hed
@@ -53,8 +58,11 @@ def hed(img, res):
     result = model_hed(img)
     return [result]
 
+
 # Pidi
 model_pidi = None
+
+
 def pidi(img, res):
     img = resize_image(HWC3(img), res)
     global model_pidi
@@ -65,8 +73,11 @@ def pidi(img, res):
     result = model_pidi(img)
     return [result]
 
+
 # MLSD
 model_mlsd = None
+
+
 def mlsd(img, res, thr_v, thr_d):
     img = resize_image(HWC3(img), res)
     global model_mlsd
@@ -77,8 +88,11 @@ def mlsd(img, res, thr_v, thr_d):
     result = model_mlsd(img, thr_v, thr_d)
     return [result]
 
+
 # Midas
 model_midas = None
+
+
 def midas(img, res):
     img = resize_image(HWC3(img), res)
     global model_midas
@@ -89,8 +103,11 @@ def midas(img, res):
     result = model_midas(img)
     return [result]
 
+
 # Zoe
 model_zoe = None
+
+
 def zoe(img, res):
     img = resize_image(HWC3(img), res)
     global model_zoe
@@ -101,8 +118,11 @@ def zoe(img, res):
     result = model_zoe(img)
     return [result]
 
+
 # Normal Bae
 model_normalbae = None
+
+
 def normalbae(img, res):
     img = resize_image(HWC3(img), res)
     global model_normalbae
@@ -113,8 +133,11 @@ def normalbae(img, res):
     result = model_normalbae(img)
     return [result]
 
+
 # DWPose
 model_dwpose = None
+
+
 def dwpose(img, res):
     img = resize_image(HWC3(img), res)
     global model_dwpose
@@ -125,8 +148,11 @@ def dwpose(img, res):
     result = model_dwpose(img)
     return [result]
 
+
 # OpenPose
 model_openpose = None
+
+
 def openpose(img, res, hand_and_face):
     img = resize_image(HWC3(img), res)
     global model_openpose
@@ -152,6 +178,8 @@ def openpose(img, res, hand_and_face):
 # Lineart
 model_lineart_anime = None
 model_lineart = None
+
+
 def lineart(img, res, preprocessor_name="Lineart", invert=True, old_images=None):
     print("Old Images: ", old_images)
     result_images = []
@@ -186,8 +214,11 @@ def lineart(img, res, preprocessor_name="Lineart", invert=True, old_images=None)
         result_images.extend(old_images)
     return result_images
 
+
 # OneFormer
 model_oneformer_coco = None
+
+
 def oneformer_coco(img, res):
     img = resize_image(HWC3(img), res)
     global model_oneformer_coco
@@ -198,8 +229,11 @@ def oneformer_coco(img, res):
     result = model_oneformer_coco(img)
     return [result]
 
+
 # OneFormer ADE20k
 model_oneformer_ade20k = None
+
+
 def oneformer_ade20k(img, res):
     img = resize_image(HWC3(img), res)
     global model_oneformer_ade20k
@@ -210,8 +244,11 @@ def oneformer_ade20k(img, res):
     result = model_oneformer_ade20k(img)
     return [result]
 
+
 # Content Shuffler
 model_content_shuffler = None
+
+
 def content_shuffler(img, res):
     img = resize_image(HWC3(img), res)
     global model_content_shuffler
@@ -222,8 +259,11 @@ def content_shuffler(img, res):
     result = model_content_shuffler(img)
     return [result]
 
+
 # Color Shuffler
 model_color_shuffler = None
+
+
 def color_shuffler(img, res):
     img = resize_image(HWC3(img), res)
     global model_color_shuffler
@@ -234,8 +274,11 @@ def color_shuffler(img, res):
     result = model_color_shuffler(img)
     return [result]
 
+
 # Inpaint
 model_inpaint = None
+
+
 def inpaint(image, invert):
     #    color = HWC3(image["image"])
     color = HWC3(image["background"])
@@ -283,12 +326,7 @@ 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=invert, outputs=invert)
-                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 Canny
                 with gr.Tab("Canny Edge") as tab_canny:
                     with gr.Row():
@@ -299,14 +337,25 @@ 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}
 
                 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"), inputs=None, outputs=all_run_btns)
-                
+
+                tab_lineart.select(
+                    fn=set_inputs(
+                        "lineart",
+                    ),
+                    inputs=all_run_btns,
+                    outputs=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")