Spaces:
Running
Running
add generate qr code command
Browse files- myapp/cli.py +10 -0
myapp/cli.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
import click
|
2 |
import dotenv
|
|
|
3 |
from huggingface_hub import InferenceClient
|
|
|
4 |
|
5 |
from myapp.palette import extract_color_clusters, generate_palette_image
|
6 |
from myapp.params import ImageParamType
|
@@ -40,3 +42,11 @@ def generate_palette(image, target, n_colors, shades):
|
|
40 |
k_means = extract_color_clusters(image, n_colors)
|
41 |
palette = generate_palette_image(k_means, shades=shades)
|
42 |
palette.save(target)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import click
|
2 |
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
|
|
|
42 |
k_means = extract_color_clusters(image, n_colors)
|
43 |
palette = generate_palette_image(k_means, shades=shades)
|
44 |
palette.save(target)
|
45 |
+
|
46 |
+
|
47 |
+
@cli.command()
|
48 |
+
@click.option("--text", required=True)
|
49 |
+
@click.option("--background", type=ImageParamType(), required=True)
|
50 |
+
@click.option("--target", type=click.Path(dir_okay=False), required=True)
|
51 |
+
def generate_qr_code(text, background, target):
|
52 |
+
write_artistic(segno.make(text), background.filename, target)
|