m3g4p0p commited on
Commit
ff91eea
·
1 Parent(s): e838dda

add error option

Browse files
Files changed (2) hide show
  1. TODO +5 -2
  2. myapp/cli.py +9 -2
TODO CHANGED
@@ -1,3 +1,6 @@
1
  Todo:
2
- ☐ Upload button
3
- Color group
 
 
 
 
1
  Todo:
2
+ Gradio:
3
+ Upload button
4
+ ☐ Color group
5
+ CLI:
6
+ ☐ Buffer context
myapp/cli.py CHANGED
@@ -3,6 +3,7 @@ import dotenv
3
  import segno
4
  from huggingface_hub import InferenceClient
5
  from qrcode_artistic import write_artistic
 
6
 
7
  from myapp.palette import extract_color_clusters, generate_palette_image
8
  from myapp.params import ImageParamType
@@ -49,5 +50,11 @@ def generate_palette(image, target, n_colors, shades):
49
  @click.option("--background", type=ImageParamType(), required=True)
50
  @click.option("--target", type=click.Path(dir_okay=False), required=True)
51
  @click.option("--scale", type=click.IntRange(min=3, max=15), default=9)
52
- def generate_qr_code(text, background, target, scale):
53
- write_artistic(segno.make(text), background.filename, target, scale=scale)
 
 
 
 
 
 
 
3
  import segno
4
  from huggingface_hub import InferenceClient
5
  from qrcode_artistic import write_artistic
6
+ from segno.consts import ERROR_MAPPING
7
 
8
  from myapp.palette import extract_color_clusters, generate_palette_image
9
  from myapp.params import ImageParamType
 
50
  @click.option("--background", type=ImageParamType(), required=True)
51
  @click.option("--target", type=click.Path(dir_okay=False), required=True)
52
  @click.option("--scale", type=click.IntRange(min=3, max=15), default=9)
53
+ @click.option("--error", type=click.Choice(list(ERROR_MAPPING)))
54
+ def generate_qr_code(text, background, target, scale, error):
55
+ write_artistic(
56
+ segno.make(text, error=error),
57
+ background.filename,
58
+ target,
59
+ scale=scale,
60
+ )