add colorutils module

myapp/app.py

@@ -10,7 +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, sort_color_clusters
+from myapp.colorutils import array_to_hex
+from myapp.palette import extract_color_clusters, sort_color_clusters
 
 try:
     import dotenv

myapp/colorutils.py

@@ -0,0 +1,33 @@
+import colorsys
+from typing import NamedTuple
+
+import numpy as np
+
+
+class HSVUpdate(NamedTuple):
+    h: int = 0
+    s: int = 0
+    v: int = 0
+
+    def apply(self, color: np.ndarray):
+        hsv = colorsys.rgb_to_hsv(*color / 255)
+        tmp = np.add(hsv, self).clip(0, 1)
+        rgb = colorsys.hsv_to_rgb(*tmp)
+
+        return np.array(rgb) * 255
+
+
+def get_hsv_value(cluster: np.ndarray):
+    return colorsys.rgb_to_hsv(*cluster / 255)[2]
+
+
+def add_hsv_saturation(cluster: np.ndarray, delta: float):
+    h, s, v = colorsys.rgb_to_hsv(*cluster / 255)
+    s = max(0, min(1, s + delta))
+
+    return np.array(colorsys.hsv_to_rgb(h, s, v)) * 255
+
+
+def array_to_hex(values: np.ndarray):
+    values = np.round(values).astype(int)
+    return "#" + ("{:02X}" * len(values)).format(*values)

myapp/palette.py

@@ -1,20 +1,10 @@
-import colorsys
 import itertools
 
 import numpy as np
 from PIL import Image
 from sklearn.cluster import KMeans
 
-
-def get_hsv_value(cluster: np.ndarray):
-    return colorsys.rgb_to_hsv(*cluster / 255)[2]
-
-
-def add_hsv_saturation(cluster: np.ndarray, delta: float):
-    h, s, v = colorsys.rgb_to_hsv(*cluster / 255)
-    s = max(0, min(1, s + delta))
-
-    return np.array(colorsys.hsv_to_rgb(h, s, v)) * 255
+from myapp.colorutils import add_hsv_saturation, get_hsv_value
 
 
 def extract_color_clusters(image_array: np.ndarray | Image.Image, n_clusters=2):
@@ -49,8 +39,3 @@ def generate_palette_image(k_means: KMeans, size=40, shades=(0.0,)):
         image.paste(part, position)
 
     return image
-
-
-def array_to_hex(values: np.ndarray):
-    values = np.round(values).astype(int)
-    return "#" + ("{:02X}" * len(values)).format(*values)

myapp/pall_app.py

@@ -3,7 +3,8 @@ from contextlib import ExitStack
 import gradio as gr
 import numpy as np
 
-from myapp.palette import array_to_hex, extract_color_clusters, iter_color_shades
+from myapp.colorutils import array_to_hex
+from myapp.palette import extract_color_clusters, iter_color_shades
 
 with gr.Blocks() as demo:
     image = gr.Image("vulture.webp")