Update ui/ui_components.py

ui/ui_components.py

@@ -1,15 +1,10 @@
 #!/usr/bin/env python3
 """
-UI Components for BackgroundFX Pro
-----------------------------------
+UI Components for BackgroundFX Pro (forced Two-Stage)
+-----------------------------------------------------
 * Pure layout: no heavy logic
 * All callbacks live in ui/callbacks.py
-* Removed first-frame video preview (redundant)
-* Improved background section with:
-  a) Upload file option
-  b) Pre-loaded professional images (dropdown)
-  c) Pre-loaded gradients & full colors (dropdown)
-  d) AI-generated background option (inline)
+* Two-stage mode is always active (checkbox removed)
 """
 
 from __future__ import annotations
@@ -40,7 +35,6 @@
 """
 
 # Keep in sync with utils/cv_processing.PROFESSIONAL_BACKGROUNDS
-# (We split existing keys into two UI groups without inventing new keys)
 _BG_CHOICES = [
     "minimalist",
     "office_modern",
@@ -165,10 +159,6 @@ def create_interface() -> gr.Blocks:
 
                     # ── Advanced options accordion ───────────────────────
                     with gr.Accordion("Advanced", open=False):
-                        use_two_stage = gr.Checkbox(
-                            label="Use Two-Stage Pipeline",
-                            value=False
-                        )
                         chroma_preset = gr.Dropdown(
                             label="Chroma Preset",
                             choices=["standard"],    # can add 'studio', 'outdoor' later
@@ -239,14 +229,14 @@ def _toggle_bg_sections(choice: str):
         # Load models
         btn_load.click(cb_load_models, outputs=statusbox)
 
-        # Process video
+        # Always two-stage: pass use_two_stage=True to callback
         btn_run.click(
             cb_process_video,
             inputs=[
                 video,
-                bg_style,        # kept for compatibility; ignored if custom_bg is set
-                custom_bg,       # becomes the effective background in all methods
-                use_two_stage,
+                bg_style,
+                custom_bg,
+                gr.State(value=True),  # Always two-stage
                 chroma_preset,
                 key_color_mode,
                 preview_mask,