Update app.py

app.py

@@ -14,6 +14,15 @@ headers = {
     "Authorization": f"Bearer {OCTOAI_TOKEN}"
 }
 
+# Function to send request to the API for background removal
+from PIL import ImageColor
+
+def hex_to_rgba(hex_color):
+    """
+    Convert a hex color string (e.g., "#ffcccb") to an RGBA tuple.
+    """
+    return ImageColor.getcolor(hex_color, "RGBA")
+
 # Function to send request to the API for background removal
 def remove_bg(image, background_choice, bg_color, bg_image):
     # Convert PIL image to base64
@@ -40,7 +49,9 @@ def remove_bg(image, background_choice, bg_color, bg_image):
 
             # Apply background based on user's choice
             if background_choice == "Solid Color":
-                background = Image.new("RGBA", removed_bg_image.size, bg_color + (255,))
+                # Convert the hex color to an RGBA tuple
+                rgba_color = hex_to_rgba(bg_color)
+                background = Image.new("RGBA", removed_bg_image.size, rgba_color)
             elif background_choice == "Image" and bg_image is not None:
                 background = bg_image.resize(removed_bg_image.size).convert("RGBA")
             else: