28. Sept. 2024, 12:43

app.py

@@ -17,12 +17,20 @@ custom_head = f"""
         <script src="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.9.0/js/all.min.js"></script>
 """
 
-title = "ControlNet v1.1 Preprocessors Standalone"
+# MARK: Gradio Theme
+theme = gr.themes.Soft(
+    primary_hue="emerald",
+    radius_size="sm",
+    neutral_hue=gr.themes.Color(c100="#a6adc8", c200="#9399b2", c300="#7f849c", c400="#6c7086", c50="#cdd6f4", c500="#585b70", c600="#45475a", c700="#313244", c800="#1e1e2e", c900="#181825", c950="#11111b"),
+)
 
+title = "ControlNet v1.1 Preprocessors Standalone"
 
-model_canny = None
 
+# MARK: Funktionen
 
+# MARK: Canny
+model_canny = None
 def canny(img, res, l, h):
     img = resize_image(HWC3(img), res)
     global model_canny
@@ -33,10 +41,8 @@ def canny(img, res, l, h):
     result = model_canny(img, l, h)
     return [result]
 
-
+# MARK: Hed
 model_hed = None
-
-
 def hed(img, res):
     img = resize_image(HWC3(img), res)
     global model_hed
@@ -47,10 +53,8 @@ def hed(img, res):
     result = model_hed(img)
     return [result]
 
-
+# MARK: Pidi
 model_pidi = None
-
-
 def pidi(img, res):
     img = resize_image(HWC3(img), res)
     global model_pidi
@@ -61,10 +65,8 @@ def pidi(img, res):
     result = model_pidi(img)
     return [result]
 
-
+# MARK: MLSD
 model_mlsd = None
-
-
 def mlsd(img, res, thr_v, thr_d):
     img = resize_image(HWC3(img), res)
     global model_mlsd
@@ -75,10 +77,8 @@ def mlsd(img, res, thr_v, thr_d):
     result = model_mlsd(img, thr_v, thr_d)
     return [result]
 
-
+# MARK: Midas
 model_midas = None
-
-
 def midas(img, res):
     img = resize_image(HWC3(img), res)
     global model_midas
@@ -89,10 +89,8 @@ def midas(img, res):
     result = model_midas(img)
     return [result]
 
-
+# MARK: Zoe
 model_zoe = None
-
-
 def zoe(img, res):
     img = resize_image(HWC3(img), res)
     global model_zoe
@@ -103,10 +101,8 @@ def zoe(img, res):
     result = model_zoe(img)
     return [result]
 
-
+# MARK: Normal Bae
 model_normalbae = None
-
-
 def normalbae(img, res):
     img = resize_image(HWC3(img), res)
     global model_normalbae
@@ -117,10 +113,8 @@ def normalbae(img, res):
     result = model_normalbae(img)
     return [result]
 
-
+# MARK: DWPose
 model_dwpose = None
-
-
 def dwpose(img, res):
     img = resize_image(HWC3(img), res)
     global model_dwpose
@@ -131,10 +125,8 @@ def dwpose(img, res):
     result = model_dwpose(img)
     return [result]
 
-
+# MARK: OpenPose
 model_openpose = None
-
-
 def openpose(img, res, hand_and_face):
     img = resize_image(HWC3(img), res)
     global model_openpose
@@ -146,9 +138,7 @@ def openpose(img, res, hand_and_face):
     return [result]
 
 
-model_uniformer = None
-
-
+# model_uniformer = None
 # def uniformer(img, res):
 #    img = resize_image(HWC3(img), res)
 #    global model_uniformer
@@ -159,12 +149,9 @@ model_uniformer = None
 #    return [result]
 
 
-## FUNCTION LINEART
-
+# MARK: 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 = []
@@ -199,10 +186,8 @@ def lineart(img, res, preprocessor_name="Lineart", invert=True, old_images=None)
         result_images.extend(old_images)
     return result_images
 
-
+# MARK: OneFormer
 model_oneformer_coco = None
-
-
 def oneformer_coco(img, res):
     img = resize_image(HWC3(img), res)
     global model_oneformer_coco
@@ -213,10 +198,8 @@ def oneformer_coco(img, res):
     result = model_oneformer_coco(img)
     return [result]
 
-
+# MARK: OneFormer ADE20k
 model_oneformer_ade20k = None
-
-
 def oneformer_ade20k(img, res):
     img = resize_image(HWC3(img), res)
     global model_oneformer_ade20k
@@ -227,10 +210,8 @@ def oneformer_ade20k(img, res):
     result = model_oneformer_ade20k(img)
     return [result]
 
-
+# MARK: Content Shuffler
 model_content_shuffler = None
-
-
 def content_shuffler(img, res):
     img = resize_image(HWC3(img), res)
     global model_content_shuffler
@@ -241,10 +222,8 @@ def content_shuffler(img, res):
     result = model_content_shuffler(img)
     return [result]
 
-
+# MARK: Color Shuffler
 model_color_shuffler = None
-
-
 def color_shuffler(img, res):
     img = resize_image(HWC3(img), res)
     global model_color_shuffler
@@ -255,10 +234,8 @@ def color_shuffler(img, res):
     result = model_color_shuffler(img)
     return [result]
 
-
+# MARK: Inpaint
 model_inpaint = None
-
-
 def inpaint(image, invert):
     #    color = HWC3(image["image"])
     color = HWC3(image["background"])
@@ -272,12 +249,6 @@ def inpaint(image, invert):
     return [result]
 
 
-theme = gr.themes.Soft(
-    primary_hue="emerald",
-    radius_size="sm",
-    neutral_hue=gr.themes.Color(c100="#a6adc8", c200="#9399b2", c300="#7f849c", c400="#6c7086", c50="#cdd6f4", c500="#585b70", c600="#45475a", c700="#313244", c800="#1e1e2e", c900="#181825", c950="#11111b"),
-)
-
 # MARK: GRADIO UI
 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"):