Minor changes to return two outputs to fix error

app.py

@@ -73,6 +73,13 @@ def hex_to_rgba(hex_color):
         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))
 
+# Define the Buy Me a Coffee button HTML (moved inside the function for consistency)
+bmc_button = """
+<a href="https://coff.ee/rdjarbeng" target="_blank">
+  <img src="https://cdn.buymeacoffee.com/buttons/v2/default-orange.png" alt="Buy Me a Coffee" style="height: 41px; width: 174px;">
+</a>
+"""
+
 def remove_background(input_path, bg_color=None, transparent_bg=True, model_choice="", alpha_matting=False, post_process_mask=False, only_mask=False):
     print(f"DEBUG: Function called with input_path: {input_path}")
     print(f"DEBUG: Parameters - bg_color: {bg_color}, transparent_bg: {transparent_bg}, model_choice: {model_choice}")
@@ -81,7 +88,7 @@ def remove_background(input_path, bg_color=None, transparent_bg=True, model_choi
         # Check if input path is valid
         if not input_path or not os.path.exists(input_path):
             print(f"ERROR: Invalid input path: {input_path}")
-            return None
+            return None, bmc_button  # Return None for image and the button HTML
             
         print("DEBUG: Starting background removal...")
         
@@ -173,17 +180,17 @@ def remove_background(input_path, bg_color=None, transparent_bg=True, model_choi
         if os.path.exists(output_path):
             file_size = os.path.getsize(output_path)
             print(f"DEBUG: Output file created successfully, size: {file_size} bytes")
-            return output_path
+            return output_path, bmc_button  # Return both the image path and the button HTML
         else:
             print("ERROR: Output file was not created")
-            return None
+            return None, bmc_button  # Return None for image and the button HTML
 
     except Exception as e:
         print(f"ERROR: Exception occurred: {str(e)}")
         import traceback
         traceback.print_exc()
         logging.error(f"An error occurred: {e}")
-        return None
+        return None, bmc_button  # Return None for image and the button HTML
 
 # Fixed examples with proper values for all inputs
 examples = [
@@ -198,12 +205,6 @@ examples = [
     ]
 ]
 
-bmc_button = """
-<a href="https://coff.ee/rdjarbeng" target="_blank">
-  <img src="https://cdn.buymeacoffee.com/buttons/v2/default-orange.png" alt="Buy Me a Coffee" style="height: 41px; width: 174px;">
-</a>
-"""
-
 # Gradio interface
 iface = gr.Interface(
     fn=remove_background,
@@ -214,7 +215,7 @@ iface = gr.Interface(
         gr.Dropdown(
             choices=[""] + [f"{k} | {v}" for k, v in MODEL_OPTIONS.items() if k != ""],
             label="Model Selection",
-            value="",  # Changed from empty to match choices
+            value="",
             allow_custom_value=False
         ),
         gr.Checkbox(label="Enable Alpha Matting", value=False),
@@ -223,7 +224,7 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.Image(type="filepath", label="Output Image (PNG)"),
-        gr.HTML(bmc_button)
+        gr.HTML(label="Support This Project")  # Added label for clarity
     ],
     examples=examples,
     title="Background Remover v2.9",