Added default parameters to the function signature, Updated examples with proper values for all 7 parameters

app.py

@@ -51,12 +51,14 @@ MODEL_OPTIONS = {
 }
 
 def hex_to_rgba(hex_color):
+    if not hex_color:
+        return None
     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_path, bg_color, transparent_bg, model_choice, alpha_matting, post_process_mask, only_mask):
+def remove_background(input_path, bg_color=None, transparent_bg=True, model_choice="", alpha_matting=False, post_process_mask=False, only_mask=False):
     try:
         # Log user interaction (minimal)
         log_user_interaction()
@@ -76,30 +78,37 @@ def remove_background(input_path, bg_color, transparent_bg, model_choice, alpha_
         session = new_session(model_name) if model_name else None
         
         # Use transparent background if selected, otherwise use color
-        bg_color_rgba = None if transparent_bg else (hex_to_rgba(bg_color) if bg_color else None)
+        bg_color_rgba = None if transparent_bg else hex_to_rgba(bg_color)
         
         # Prepare additional options
         remove_kwargs = {
             "session": session,
-            "bgcolor": bg_color_rgba,
-            "alpha_matting": alpha_matting,
-            "post_process_mask": post_process_mask,
-            "only_mask": only_mask
         }
-
-        # Add alpha matting parameters if enabled
+        
+        # Only add bgcolor if we have one
+        if bg_color_rgba is not None:
+            remove_kwargs["bgcolor"] = bg_color_rgba
+            
+        # Add other parameters
         if alpha_matting:
             remove_kwargs.update({
+                "alpha_matting": True,
                 "alpha_matting_foreground_threshold": 270,
                 "alpha_matting_background_threshold": 20,
                 "alpha_matting_erode_size": 11
             })
+        
+        if post_process_mask:
+            remove_kwargs["post_process_mask"] = True
+            
+        if only_mask:
+            remove_kwargs["only_mask"] = True
 
         # Convert PIL Image to numpy array
         input_array = np.array(input_image)
 
         # Use the remove function
-        output_array = remove(input_array, **{k: v for k, v in remove_kwargs.items() if v is not None})
+        output_array = remove(input_array, **remove_kwargs)
 
         # Convert numpy array back to PIL Image
         output_image = Image.fromarray(output_array)
@@ -122,12 +131,13 @@ def remove_background(input_path, bg_color, transparent_bg, model_choice, alpha_
         logging.error(f"An error occurred: {e}")
         return None
 
+# Fixed examples with proper values for all inputs
 examples = [
     [
         'scifi_man1.jpg',  # input_path
-        None,              # bg_color
-        True,              # transparent_bg (most common use case)
-        "",                # model_choice
+        "#FFFFFF",         # bg_color (white)
+        True,              # transparent_bg
+        "",                # model_choice (empty string, which is in the choices)
         False,             # alpha_matting
         False,             # post_process_mask
         False              # only_mask
@@ -139,12 +149,13 @@ iface = gr.Interface(
     fn=remove_background,
     inputs=[
         gr.Image(type="filepath", label="Input Image"),
-        gr.ColorPicker(label="Background Color (ignored if transparent is selected)", value=None),
-        gr.Checkbox(label="Transparent Background", value=False),
+        gr.ColorPicker(label="Background Color (ignored if transparent is selected)", value="#FFFFFF"),
+        gr.Checkbox(label="Transparent Background", value=True),
         gr.Dropdown(
             choices=[""] + [f"{k} | {v}" for k, v in MODEL_OPTIONS.items() if k != ""],
             label="Model Selection",
-            value=""
+            value="",  # Changed from empty to match choices
+            allow_custom_value=False
         ),
         gr.Checkbox(label="Enable Alpha Matting", value=False),
         gr.Checkbox(label="Post-Process Mask", value=False),
@@ -160,4 +171,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=True)  # Added share=True for Hugging Face deployment