update char images

app.py

@@ -1,30 +1,50 @@
+import os
+import random
+
 import gradio as gr
 import numpy as np
 
 from ocr import Recognizer
 
-with gr.Blocks() as app:
+chars = [os.path.join(dn, fn) for dn, _, ff in os.walk("chars") for fn in ff]
+
+font = gr.themes.GoogleFont("Noto Sans")
+theme = gr.themes.Soft(font=font)
+
+with gr.Blocks(theme=theme, title="Kana Writer") as app:
     recog = Recognizer("model/model.xml", "model/char_list.txt")
 
+    brush = gr.Brush(default_color="#111", default_size=15)
+    sketch = gr.Sketchpad(
+        random.choice(chars),
+        type="numpy",
+        brush=brush,
+        layers=False,
+        image_mode="RGB",
+        label="寫字練習",
+    )
+    result = gr.TextArea(label="辨識結果", lines=1)
+
     with gr.Row():
-        brush = gr.Brush(default_color="#FFF", default_size=15)
-        sketch = gr.Sketchpad("bg.png", type="numpy", brush=brush, layers=False)
-        result = gr.TextArea()
-    check = gr.Button()
+        check_btn = gr.Button("辨識")
+        rand_btn = gr.Button("隨機")
 
     def parse_item(item):
         prob = item["prob"]
         char = item["char"]
-        return f"{char}: {prob}"
+        return f"{char}: {prob:.2%}"
 
     def do_recog(img: dict[str, np.ndarray]):
-        _, nbest = recog(img["composite"])
-        return "\n".join(parse_item(i) for item in nbest for i in item)
+        img: np.ndarray = img["layers"][0]
+        img[img == 0] = 255
+        img[img != 255] = 0
+        _, nbest = recog(img)
+        return "\n".join(parse_item(i) for item in nbest for i in item if i["prob"] > 1e-2)
 
-    def clear():
-        return "bg.png"
+    def rand_char():
+        return random.choice(chars)
 
-    check.click(do_recog, sketch, result)
-    sketch.clear(clear, None, sketch)
+    check_btn.click(do_recog, sketch, result)
+    rand_btn.click(rand_char, None, sketch)
 
 app.launch()