add color pickers

myapp/app.py

@@ -1,3 +1,4 @@
+import re
 from functools import partial
 from pathlib import Path
 from tempfile import NamedTemporaryFile
@@ -33,13 +34,18 @@ with gr.Blocks() as demo:
             prompt = gr.TextArea("A psychedelic vulture", label="Prompt")
             model = gr.Radio(MODELS, value=MODELS[0], label="Model")
             button = gr.Button("Generate")
+
         with gr.Column():
+            output = gr.Image()
             background = gr.Image(visible=False, type="filepath")
             scale = gr.Slider(3, 15, 9, step=1, label="Scale")
-            output = gr.Image()
+
+            with gr.Row():
+                color_dark = gr.ColorPicker("#000000", label="Dark")
+                color_light = gr.ColorPicker("#FFFFFF", label="Light")
 
     def generate_background(data: dict[Component, Any]):
-        if prompt not in data:
+        if not data.get(prompt):
             return gr.skip(), gr.skip()
 
         return client.text_to_image(data[prompt], model=data[model]), None
@@ -48,6 +54,15 @@ with gr.Blocks() as demo:
         if data.get(background) is None:
             return None
 
+        def to_hex_format(value: str):
+            if value.startswith("#"):
+                return value
+
+            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()
+
         qr_code = segno.make(data[text], error="h")
         suffix = Path(data[background]).suffix
 
@@ -57,6 +72,8 @@ with gr.Blocks() as demo:
                 target=temp_file,
                 background=data[background],
                 scale=data[scale],
+                light=to_hex_format(data[color_light]),
+                dark=to_hex_format(data[color_dark]),
             )
 
             return Image.open(temp_file.name)
@@ -75,9 +92,21 @@ with gr.Blocks() as demo:
     )
 
     gr.on(
-        [text.submit, background.change, scale.change],
+        [
+            text.submit,
+            background.change,
+            scale.change,
+            color_light.change,
+            color_dark.change,
+        ],
         generate_output,
-        inputs={text, background, scale},
+        inputs={
+            text,
+            background,
+            scale,
+            color_light,
+            color_dark,
+        },
         outputs=output,
     )