Run Black formatter

app.py

@@ -28,7 +28,16 @@ def ensure_image(img) -> Image.Image:
         raise ValueError("Cannot convert input to a PIL Image.")
 
 
-def call_baseten_generate(image: Image.Image, prompt: str, steps: int, strength: float, height: int, width: int, lora_name: str, remove_bg: bool) -> Image.Image | None:
+def call_baseten_generate(
+    image: Image.Image,
+    prompt: str,
+    steps: int,
+    strength: float,
+    height: int,
+    width: int,
+    lora_name: str,
+    remove_bg: bool,
+) -> Image.Image | None:
     image = ensure_image(image)
     b64_image = image_to_base64(image)
     payload = {
@@ -63,15 +72,18 @@ def call_baseten_generate(image: Image.Image, prompt: str, steps: int, strength:
 
 # ================== MODE CONFIG =====================
 
-Mode = TypedDict("Mode", {
-    "model": str,
-    "prompt": str,
-    "default_strength": float,
-    "default_height": int,
-    "default_width": int,
-    "models": list[str],
-    "remove_bg": bool,
-})
+Mode = TypedDict(
+    "Mode",
+    {
+        "model": str,
+        "prompt": str,
+        "default_strength": float,
+        "default_height": int,
+        "default_width": int,
+        "models": list[str],
+        "remove_bg": bool,
+    },
+)
 
 MODE_DEFAULTS: dict[str, Mode] = {
     "Subject Generation": {
@@ -125,14 +137,21 @@ MODE_DEFAULTS: dict[str, Mode] = {
 
 MODE_EXAMPLES = {
     "Subject Generation": [
-        ["assets/subj1.jpg", "Close-up portrait of a fruit bowl", "assets/subj1_out.jpg"],
+        [
+            "assets/subj1.jpg",
+            "Close-up portrait of a fruit bowl",
+            "assets/subj1_out.jpg",
+        ],
         ["assets/subj2.jpg", "A penguin standing in snow", "assets/subj2_out.jpg"],
         ["assets/subj3.jpg", "A cat with glowing eyes", "assets/subj3_out.jpg"],
         ["assets/subj4.jpg", "A child playing with bubbles", "assets/subj4_out.jpg"],
-        ["assets/subj5.jpg", "A stylish young man in neon lights", "assets/subj5_out.jpg"],
+        [
+            "assets/subj5.jpg",
+            "A stylish young man in neon lights",
+            "assets/subj5_out.jpg",
+        ],
         ["assets/subj6.jpg", "Old man with a mysterious look", "assets/subj6_out.jpg"],
     ],
-
     "Background Generation": [
         ["assets/bg1.jpg", "Modern living room with plants", "assets/bg1_out.jpg"],
         ["assets/bg2.jpg", "Fantasy forest background", "assets/bg2_out.jpg"],
@@ -141,25 +160,58 @@ MODE_EXAMPLES = {
         ["assets/bg5.jpg", "Snowy mountain landscape", "assets/bg5_out.jpg"],
         ["assets/bg6.jpg", "Golden sunset over the sea", "assets/bg6_out.jpg"],
     ],
-
     "Canny": [
         ["assets/canny1.jpg", "A neon cyberpunk city skyline", "assets/canny1_out.jpg"],
         ["assets/canny2.jpg", "A robot walking in the fog", "assets/canny2_out.jpg"],
-        ["assets/canny3.jpg", "A futuristic vehicle parked under a bridge", "assets/canny3_out.jpg"],
-        ["assets/canny4.jpg", "Sci-fi lab interior with glowing machinery", "assets/canny4_out.jpg"],
-        ["assets/canny5.jpg", "A portrait of a woman outlined in neon", "assets/canny5_out.jpg"],
-        ["assets/canny6.jpg", "Post-apocalyptic abandoned street", "assets/canny6_out.jpg"],
+        [
+            "assets/canny3.jpg",
+            "A futuristic vehicle parked under a bridge",
+            "assets/canny3_out.jpg",
+        ],
+        [
+            "assets/canny4.jpg",
+            "Sci-fi lab interior with glowing machinery",
+            "assets/canny4_out.jpg",
+        ],
+        [
+            "assets/canny5.jpg",
+            "A portrait of a woman outlined in neon",
+            "assets/canny5_out.jpg",
+        ],
+        [
+            "assets/canny6.jpg",
+            "Post-apocalyptic abandoned street",
+            "assets/canny6_out.jpg",
+        ],
     ],
-
     "Depth": [
-        ["assets/depth1.jpg", "A narrow alleyway with deep perspective", "assets/depth1_out.jpg"],
-        ["assets/depth2.jpg", "A mountain road vanishing into the distance", "assets/depth2_out.jpg"],
-        ["assets/depth3.jpg", "A hallway with strong depth of field", "assets/depth3_out.jpg"],
-        ["assets/depth4.jpg", "A misty forest path stretching far away", "assets/depth4_out.jpg"],
+        [
+            "assets/depth1.jpg",
+            "A narrow alleyway with deep perspective",
+            "assets/depth1_out.jpg",
+        ],
+        [
+            "assets/depth2.jpg",
+            "A mountain road vanishing into the distance",
+            "assets/depth2_out.jpg",
+        ],
+        [
+            "assets/depth3.jpg",
+            "A hallway with strong depth of field",
+            "assets/depth3_out.jpg",
+        ],
+        [
+            "assets/depth4.jpg",
+            "A misty forest path stretching far away",
+            "assets/depth4_out.jpg",
+        ],
         ["assets/depth5.jpg", "A bridge over a deep canyon", "assets/depth5_out.jpg"],
-        ["assets/depth6.jpg", "An underground tunnel with receding arches", "assets/depth6_out.jpg"],
+        [
+            "assets/depth6.jpg",
+            "An underground tunnel with receding arches",
+            "assets/depth6_out.jpg",
+        ],
     ],
-
     "Deblurring": [
         ["assets/deblur1.jpg", "", "assets/deblur1_out.jpg"],
         ["assets/deblur2.jpg", "", "assets/deblur2_out.jpg"],
@@ -242,8 +294,19 @@ with gr.Blocks(title="🌍 ZenCtrl") as demo:
                             label="Width",
                         )
 
-                def on_generate_click(model_name, prompt, steps, strength, height, width, remove_bg, image):
-                    return call_baseten_generate(image, prompt, steps, strength, height, width, model_name, remove_bg)
+                def on_generate_click(
+                    model_name, prompt, steps, strength, height, width, remove_bg, image
+                ):
+                    return call_baseten_generate(
+                        image,
+                        prompt,
+                        steps,
+                        strength,
+                        height,
+                        width,
+                        model_name,
+                        remove_bg,
+                    )
 
                 generate_button.click(
                     fn=on_generate_click,