Improve color parsing

app.py

@@ -10,7 +10,8 @@ import cairo
 import numpy as np
 import gradio as gr
 from pydub import AudioSegment
-
+import re
+import colorsys
 
 def read_audio(audio, seek=None, duration=None):
     """
@@ -205,16 +206,48 @@ def parse_color(colorstr):
             "Format for color is 3 floats separated by commas 0.xx,0.xx,0.xx, rgb order"
         )
 
-
-def hex_to_rgb(hex_color):
-    if hex_color.startswith("rgb"):
-        raise ValueError(f"Unexpected color format: {hex_color}")
-
-    hex_color = hex_color.lstrip('#')
-    r = int(hex_color[0:2], 16) / 255.0
-    g = int(hex_color[2:4], 16) / 255.0
-    b = int(hex_color[4:6], 16) / 255.0
-    return (r, g, b)
+def hex_to_rgb(color):
+    """
+    Convert color codes to RGB format.
+    
+    Supports:
+    - HEX codes with # (e.g., "#FFA07A")
+    - HEX codes without # (e.g., "FFA07A")
+    - RGB format (e.g., "rgb(255, 160, 122)")
+    - HSV format (e.g., "hsv(17, 52, 100)")
+
+    Returns:
+        (r, g, b): Tuple of float values (0-1)
+    """
+    print(f"Received color: {color}")  # Debugging line
+
+    # HEX with or without #
+    if color.startswith("#") or (len(color) in [6, 3] and all(c in "0123456789ABCDEFabcdef" for c in color)):
+        color = color.lstrip("#")  # Remove # if present
+
+        # Support 3-digit HEX (e.g., "FA5" -> "FFAA55")
+        if len(color) == 3:
+            color = "".join([c * 2 for c in color])
+
+        if len(color) == 6:
+            return (int(color[0:2], 16) / 255.0,
+                    int(color[2:4], 16) / 255.0,
+                    int(color[4:6], 16) / 255.0)
+
+    # RGB format (rgb(r, g, b))
+    match_rgb = re.match(r"rgb\((\d+),\s*(\d+),\s*(\d+)\)", color)
+    if match_rgb:
+        r, g, b = map(int, match_rgb.groups())
+        return (r / 255.0, g / 255.0, b / 255.0)
+
+    # HSV format (hsv(h, s, v))
+    match_hsv = re.match(r"hsv\((\d+),\s*(\d+),\s*(\d+)\)", color)
+    if match_hsv:
+        h, s, v = map(int, match_hsv.groups())
+        r, g, b = colorsys.hsv_to_rgb(h / 360.0, s / 100.0, v / 100.0)
+        return (r, g, b)
+
+    raise ValueError(f"Invalid color format: {color}")
 
 
 def do_viz(