Update app.py

app.py

@@ -4,41 +4,6 @@ import numpy as np
 from typing import TypedDict
 
 
-class ImageData(TypedDict):
-    background: np.ndarray
-    layers: list[np.ndarray]
-
-
-def mask(image: ImageData):
-    bg = image.get("background", None)
-    layers = image.get("layers", [])
-
-    if bg is None and not layers:
-        raise ValueError("No background or layers provided")
-
-    mask = layers[0]
-    for layer in layers[1:]:
-        mask = combine_masks(mask, layer)
-
-    # save images or use them elsewhere
-    # Image.fromarray(mask).save("mask.png")
-    # Image.fromarray(bg).save("bg.png")
-
-    mask = make_grey(mask)
-
-    return [bg, mask]
-
-
-with gr.Blocks() as demo:
-    imed = gr.ImageEditor()
-    im_1 = gr.Image()
-    im_2 = gr.Image()
-    imed.apply(mask, inputs=imed, outputs=[im_1, im_2])
-
-if __name__ == "__main__":
-    demo.launch()
-
-
 def combine_masks(im_1: np.ndarray, im_2: np.ndarray):
     """
     Layers images on top of each other, with the foreground image overwriting the background.
@@ -78,3 +43,39 @@ def make_grey(image: np.ndarray, grey_value: int = 128):
 
     # combine channels and return
     return np.dstack((rgb, alpha))
+    
+class ImageData(TypedDict):
+    background: np.ndarray
+    layers: list[np.ndarray]
+
+
+def mask(image: ImageData):
+    bg = image.get("background", None)
+    layers = image.get("layers", [])
+
+    if bg is None and not layers:
+        raise ValueError("No background or layers provided")
+
+    mask = layers[0]
+    for layer in layers[1:]:
+        mask = combine_masks(mask, layer)
+
+    # save images or use them elsewhere
+    # Image.fromarray(mask).save("mask.png")
+    # Image.fromarray(bg).save("bg.png")
+
+    mask = make_grey(mask)
+
+    return [bg, mask]
+
+
+with gr.Blocks() as demo:
+    imed = gr.ImageEditor()
+    im_1 = gr.Image()
+    im_2 = gr.Image()
+    imed.apply(mask, inputs=imed, outputs=[im_1, im_2])
+
+if __name__ == "__main__":
+    demo.launch()
+
+