add image param type

myapp/__main__.py

@@ -0,0 +1,4 @@
+from myapp.cli import cli
+
+if __name__ == "__main__":
+    cli()

myapp/app.py

@@ -65,7 +65,7 @@ with gr.Blocks() as demo:
             matches = re.findall(r"\d+(?:\.\d+)?", value)
             r, g, b = map(int, map(float, matches[:3]))
 
-            return f"#{r:x}{g:x}{b:x}".upper()
+            return f"#{r:02X}{g:02X}{b:02X}"
 
         image = Image.open(data[background])
         qr_code = segno.make(data[text], error="h")

myapp/cli.py

@@ -2,6 +2,8 @@ import click
 import dotenv
 from huggingface_hub import InferenceClient
 
+from myapp.types import ImageParamType
+
 dotenv.load_dotenv()
 client = InferenceClient()
 
@@ -28,5 +30,7 @@ def generate_image(prompt, target, model, width, height):
     image.save(target)
 
 
-if __name__ == "__main__":
-    cli()
+@cli.command()
+@click.option("--image", type=ImageParamType(), required=True)
+def generate_palette(image):
+    pass

myapp/types.py

@@ -0,0 +1,19 @@
+from typing import Any
+
+import click
+from PIL import Image
+
+
+class ImageParamType(click.ParamType):
+    name = "image"
+
+    def convert(
+        self, value: Any, param: click.Parameter | None, ctx: click.Context | None
+    ) -> Any:
+        if value is None or isinstance(value, Image.Image):
+            return value
+
+        try:
+            return Image.open(value)
+        except OSError as e:
+            self.fail(str(e))