Update utils.py

utils.py

@@ -39,19 +39,24 @@ TRANSPARENT_BG = dict(
 # Speaker sample helpers
 # ---------------------------------------------------------------------------
 
-def colorsCSS(n, startingHue=None):
+def colorsCSS(n, startingHue=None, pool=None):
     """Creates n distinctive CSS hex colors evenly spaced across the hue wheel.
 
     Drop-in replacement for su.colorsCSS that accepts an optional startingHue
     so related color pools (catTypeColors, speakerColors, catColors) can share
     the same starting point and stay visually consistent across rerenders.
+
+    pool: if set, generate this many evenly-spaced colors first, then pick n
+    from them. This constrains the color space so colors are never too similar
+    even when n is small. Recommended value: 24.
     """
     if n == 0:
         return []
-    ret = []
     h = startingHue if startingHue is not None else int(random.random() * 180)
-    step = 180 / n
-    for _ in range(n):
+    num = pool if (pool and pool >= n) else n
+    step = 180 / num
+    all_colors = []
+    for _ in range(num):
         h += step
         h = int(h) % 180
         hsv = np.uint8([[[h, 200, 200]]])
@@ -59,8 +64,10 @@ def colorsCSS(n, startingHue=None):
         b = f'{bgr[0][0][0].item():02x}'
         g = f'{bgr[0][0][1].item():02x}'
         r = f'{bgr[0][0][2].item():02x}'
-        ret.append('#' + b + g + r)
-    return ret
+        all_colors.append('#' + b + g + r)
+    # Pick n evenly-spaced colors from the pool
+    indices = [int(i * num / n) for i in range(n)]
+    return [all_colors[i] for i in indices]
 
 
 def extract_clip_bytes(waveform, sample_rate, seg_start, seg_end):