use split method

app.py

@@ -27,8 +27,11 @@ 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))
 
-def remove_background(input_image, bg_color, model_name, alpha_matting, post_process_mask, only_mask):
+def remove_background(input_image, bg_color, model_choice, alpha_matting, post_process_mask, only_mask):
     try:
+        # Extract the model name from the choice
+        model_name = model_choice.split(' | ')[0] if model_choice else ""
+        
         # 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
         
@@ -71,7 +74,7 @@ def remove_background(input_image, bg_color, model_name, alpha_matting, post_pro
         # Save the output image to a temporary file
         # Generate a unique timestamp for the output file name
         timestamp = time.strftime("%Y%m%d-%H%M%S")
-        output_path = f"output_remove_background_s{timestamp}.png"
+        output_path = f"output_remove_background_{timestamp}.png"
         output_image.save(output_path)
         logging.info(f"Saved output image {output_path}")
 
@@ -87,11 +90,12 @@ iface = gr.Interface(
     inputs=[
         gr.Image(type="pil"),
         gr.ColorPicker(label="Background Color", value=None),
-        gr.Dropdown(choices=list(MODEL_OPTIONS.keys()),
-                    label="Model Selection",
-                    value="",
-                    type="value",
-                    info=lambda x: MODEL_OPTIONS[x] if x in MODEL_OPTIONS else ""),
+        gr.Dropdown(
+            choices=[""] + [f"{k} | {v}" for k, v in MODEL_OPTIONS.items() if k != ""],
+            label="Model Selection",
+            value="",
+            type="value"
+        ),
         gr.Checkbox(label="Enable Alpha Matting", value=False),
         gr.Checkbox(label="Post-Process Mask (to post process the mask to get better results)", value=False),
         gr.Checkbox(label="Only Return Mask ", value=False)
@@ -100,7 +104,7 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Output Image"), 
         gr.File(label="Download the output image")
     ],
-    title="Advanced Background Remover v2.4",
+    title="Advanced Background Remover v2.5",
     description="Upload an image to remove the background. Customize the result with different options, including background color, model selection, alpha matting, and more.",
     allow_flagging="never",
 )