add hex to rgba, fix for bg color

app.py

@@ -6,22 +6,31 @@ import logging
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 
+def hex_to_rgba(hex_color):
+    hex_color = hex_color.lstrip('#')
+    if len(hex_color) == 6:
+        hex_color += 'FF'  # Add full opacity if no alpha is provided
+    return tuple(int(hex_color[i:i + 2], 16) for i in (0, 2, 4, 6))
+
 def remove_background(input_image, bg_color, model_name, alpha_matting, post_process_mask, only_mask):
     try:
         # Set up the session with the chosen model, or None if no model is selected
         session = new_session(model_name) if model_name else None
         
+        # Convert hex color to RGBA tuple
+        bg_color_rgba = hex_to_rgba(bg_color) if bg_color else None
+        
         # Prepare additional options
         remove_kwargs = {
             "session": session,
-            "bgcolor": bg_color if bg_color else None,
+            "bgcolor": bg_color_rgba if bg_color_rgba else None,
             "alpha_matting": alpha_matting,
             "post_process_mask": post_process_mask,
             "only_mask": only_mask
         }
         
         # Use the remove function
-        if session:
+        if session or bg_color_rgba:
             output_image = remove(input_image, **{k: v for k, v in remove_kwargs.items() if v is not None})
         else:
             output_image = remove(input_image)  # Use the default remove function