Update app.py

app.py

@@ -14,9 +14,16 @@ from transparent_background import Remover
 @spaces.GPU()
 def doo(video, color, mode, progress=gr.Progress()):
     print(color)
-    color = color.lstrip('#') 
-    rgb = tuple(int(color[i:i+2], 16) for i in (0, 2, 4))
-    color = str(list(rgb))
+    if color.startswith('#'):
+        color = color.lstrip('#')
+        rgb = tuple(int(color[i:i+2], 16) for i in (0, 2, 4))
+        color = str(list(rgb))
+    elif color.startswith('rgba'):
+        rgba_match = re.match(r'rgba\((\d+), (\d+), (\d+), ([\d.]+)\)', color)
+        if rgba_match:
+            r, g, b, _ = rgba_match.groups()
+            color = str([int(r), int(g), int(b)])
+    print(color)
     if mode == 'Fast':
         remover = Remover(mode='fast')
     else: