First Commit

.gitignore

@@ -0,0 +1 @@
+__pycache__/

app.py

@@ -0,0 +1,8 @@
+from backend import SengaFiller
+import frontend
+
+from jgdildsengafiller import JGDILDSengaFiller
+
+frontend.init(
+    sengafiller=JGDILDSengaFiller()
+)

backend.py

@@ -0,0 +1,8 @@
+from abc import ABC,abstractmethod
+from numpy import ndarray
+from PIL import Image
+
+class SengaFiller(ABC):
+    @abstractmethod
+    def run(image:Image.Image)->ndarray:
+        pass

dummysengafiller.py

@@ -0,0 +1,6 @@
+from backend import SengaFiller
+from numpy import ndarray,asarray
+from PIL import Image
+class DummySengaFiller(SengaFiller):
+    def run(self,image: Image.Image) -> ndarray:
+        return asarray(image)

frontend.py

@@ -0,0 +1,48 @@
+from backend import SengaFiller
+def init(sengafiller:SengaFiller):
+    from gradio import Blocks, Markdown, Image, Row, Button, Box
+
+    with Blocks() as app:
+
+        # Prepare Components
+        Markdown(
+            """# SengaFiller
+        Connects the lines you draw so that you can fill your drawing correctly.
+        """
+        )
+        with Box().style(rounded=True, margin=True):
+            input_image = Image(label="input",image_mode="L",type="pil")
+            with Box().style(border=False):
+                with Row().style(equal_height=True):
+                    submit_button = Button("RUN", variant="primary").style(
+                        full_width=True, rounded=(True, False, False, True)
+                    )
+                    clear_button = Button("CLEAR").style(
+                        full_width=True, rounded=(False, True, True, False)
+                    )
+            output_image = Image(label="output")
+        Markdown(
+            """
+        ### Credit
+        The model `model1.h5` is licensed under a CC-BY-NC-SA 4.0 international license, created by [hepesu](https://github.com/hepesu) and available on [Release Page of LineCloser Repo](https://github.com/hepesu/LineCloser/releases)
+        """
+        )
+
+        # Event Handlers
+        def on_submit_button_click(input_image_data):
+            return sengafiller.run(input_image_data)
+        def on_clear_button_click():
+            return None,None
+
+        # Connect Components
+        submit_button.click(
+            fn=on_submit_button_click, inputs=[input_image], outputs=[output_image]
+        )
+        clear_button.click(
+            fn=on_clear_button_click,inputs=[],outputs=[input_image,output_image]
+        )
+    app.launch()
+
+
+if __name__ == "__main__":
+    init()

jgdildsengafiller.py

@@ -0,0 +1,21 @@
+from backend import SengaFiller
+from numpy import ndarray, cast, int8, ones
+from PIL import Image
+
+
+class JGDILDSengaFiller(SengaFiller):
+    def __init__(self) -> None:
+        super().__init__()
+        from tensorflow import keras
+
+        self.model = keras.models.load_model("./model1.h5")
+
+    def run(self, image_pil: Image.Image) -> ndarray:
+        input_width, input_height = image_pil.size
+        image_mono_pil = image_pil.point(lambda x: int(x > 200), mode="L")
+        image_numpy = ones((input_height, input_width)) * image_mono_pil
+        x = image_numpy.reshape((1, input_height, input_width, 1))
+        y = self.model(x)
+        output_height, output_width = y.shape[1:3]
+        output_image = y.numpy().reshape(output_height, output_width)
+        return output_image

model1.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7f60bed0cfff29523a9fedc9b8b99460f313da39ce446a1d3fb7992ded6e29d
+size 66259544

requirements.txt

@@ -0,0 +1,2 @@
+numpy
+tensorflow