m3g4p0p commited on
Commit
b3e6b4d
·
1 Parent(s): 4e87ee2

add generate qr code command

Browse files
Files changed (1) hide show
  1. 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)