rename module

myapp/cli.py

@@ -3,7 +3,7 @@ import dotenv
 from huggingface_hub import InferenceClient
 
 from myapp.palette import extract_color_clusters, generate_palette_image
-from myapp.types import ImageParamType
+from myapp.params import ImageParamType
 
 dotenv.load_dotenv()
 client = InferenceClient()

myapp/cli/main.py

@@ -0,0 +1,42 @@
+import click
+import dotenv
+from huggingface_hub import InferenceClient
+
+from myapp.palette import extract_color_clusters, generate_palette_image
+from myapp.params import ImageParamType
+
+dotenv.load_dotenv()
+client = InferenceClient()
+
+
+@click.group()
+def cli():
+    pass
+
+
+@cli.command()
+@click.option("--prompt", required=True)
+@click.option("--target", type=click.Path(dir_okay=False), required=True)
+@click.option("--model", default="black-forest-labs/FLUX.1-schnell")
+@click.option("--width", default=400)
+@click.option("--height", default=400)
+def generate_image(prompt, target, model, width, height):
+    image = client.text_to_image(
+        prompt=prompt,
+        model=model,
+        width=width,
+        height=height,
+    )
+
+    image.save(target)
+
+
+@cli.command()
+@click.option("--image", type=ImageParamType(), required=True)
+@click.option("--target", type=click.Path(dir_okay=False), required=True)
+@click.option("--n-colors", default=4)
+@click.option("--shade", "shades", default=(0.0,), multiple=True)
+def generate_palette(image, target, n_colors, shades):
+    k_means = extract_color_clusters(image, n_colors)
+    palette = generate_palette_image(k_means, shades=shades)
+    palette.save(target)

myapp/pall_app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import numpy as np
 
-from myapp.palette import array_to_hex, extract_color_clusters
+from myapp.palette import array_to_hex, extract_color_clusters, iter_color_shades
 
 with gr.Blocks() as demo:
     image = gr.Image("vulture.webp")
@@ -11,6 +11,7 @@ with gr.Blocks() as demo:
     @gr.render(inputs=[image, n_colors])
     def render_palette(image_array: np.ndarray, n_clusers: int):
         model = extract_color_clusters(image_array, n_clusers)
+        shades = iter_color_shades(model, (0, 0.2, 0.4, 0.6))
 
         for cluster in model.cluster_centers_:
             gr.ColorPicker(array_to_hex(cluster))