Spaces:
Running
Running
add error option
Browse files- TODO +5 -2
- myapp/cli.py +9 -2
TODO
CHANGED
@@ -1,3 +1,6 @@
|
|
1 |
Todo:
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
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 |
-
|
53 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
)
|