add generate palette

myapp/app.py

@@ -1,7 +1,7 @@
 import re
 from functools import partial
 from io import BytesIO
-from typing import Any
+from typing import Any, cast
 
 import gradio as gr
 import segno
@@ -10,6 +10,8 @@ from huggingface_hub import InferenceClient
 from PIL import Image
 from qrcode_artistic import write_artistic
 
+from myapp.palette import array_to_hex, extract_color_clusters
+
 try:
     import dotenv
 
@@ -54,6 +56,9 @@ with gr.Blocks() as demo:
             return None
 
         def to_hex_format(value: str):
+            if value is None:
+                return None
+
             if value.startswith("#"):
                 return value
 
@@ -69,15 +74,26 @@ with gr.Blocks() as demo:
             write_artistic(
                 qr_code,
                 target=buffer,
-                background=image.fp,
+                background=image.filename,
                 kind=image.format,
                 scale=data[scale],
                 light=to_hex_format(data[color_light]),
                 dark=to_hex_format(data[color_dark]),
+                quiet_zone=cast(Any, "#FFFFFF"),
             )
 
             return Image.open(buffer)
 
+    def generate_palette(data: dict[Component, Any]):
+        if data[background] is None:
+            return None, None
+
+        image = Image.open(data[background])
+        k_means = extract_color_clusters(image, n_clusters=2)
+        primary, secondary = map(array_to_hex, k_means.cluster_centers_)
+
+        return primary, secondary
+
     gr.on(
         [button.click, prompt.submit],
         partial(gr.update, interactive=False),
@@ -110,6 +126,12 @@ with gr.Blocks() as demo:
         outputs=output,
     )
 
+    gr.on(
+        fn=generate_palette,
+        inputs={background},
+        outputs=[color_dark, color_light],
+    )
+
 
 if __name__ == "__main__":
     demo.launch()