penut85420 commited on
Commit
ba0dd03
1 Parent(s): 399b764

adjust sketch clear behavior

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -12,11 +12,13 @@ font = gr.themes.GoogleFont("Noto Sans")
12
  theme = gr.themes.Soft(font=font)
13
 
14
  with gr.Blocks(theme=theme, title="Kana Writer") as app:
 
 
15
  recog = Recognizer("model/model.xml", "model/char_list.txt")
16
 
17
  brush = gr.Brush(default_color="#111", default_size=15)
18
  sketch = gr.Sketchpad(
19
- random.choice(chars),
20
  type="numpy",
21
  brush=brush,
22
  layers=False,
@@ -42,9 +44,14 @@ with gr.Blocks(theme=theme, title="Kana Writer") as app:
42
  return "\n".join(parse_item(i) for item in nbest for i in item if i["prob"] > 1e-2)
43
 
44
  def rand_char():
45
- return random.choice(chars)
 
46
 
47
- check_btn.click(do_recog, sketch, result)
48
- rand_btn.click(rand_char, None, sketch)
 
 
 
 
49
 
50
  app.launch()
 
12
  theme = gr.themes.Soft(font=font)
13
 
14
  with gr.Blocks(theme=theme, title="Kana Writer") as app:
15
+ char = random.choice(chars)
16
+ curr_char = gr.State(char)
17
  recog = Recognizer("model/model.xml", "model/char_list.txt")
18
 
19
  brush = gr.Brush(default_color="#111", default_size=15)
20
  sketch = gr.Sketchpad(
21
+ char,
22
  type="numpy",
23
  brush=brush,
24
  layers=False,
 
44
  return "\n".join(parse_item(i) for item in nbest for i in item if i["prob"] > 1e-2)
45
 
46
  def rand_char():
47
+ char = random.choice(chars)
48
+ return char, char
49
 
50
+ def clear(curr_char):
51
+ return curr_char
52
+
53
+ check_btn.click(do_recog, sketch, result, show_progress="minimal")
54
+ rand_btn.click(rand_char, None, [sketch, curr_char], show_progress="minimal")
55
+ sketch.clear(clear, curr_char, sketch, show_progress="minimal")
56
 
57
  app.launch()