Update ui/callbacks.py

ui/callbacks.py

@@ -3,11 +3,7 @@
 Callbacks for BackgroundFX Pro UI
 ---------------------------------
 All functions here are *thin* wrappers wired to the Gradio interface.
-
-• key_color_mode parameter already added (matches ui_components.py)
-• PREVIEW FUNCTIONS NOW IMPLEMENTED:
-    - cb_video_changed        → returns first frame
-• SIMPLIFIED: Background handled by gr.Image component directly
+NO IMPORTS FROM core.app AT MODULE LEVEL to avoid circular imports
 """
 
 from __future__ import annotations
@@ -16,14 +12,8 @@
 import numpy as np
 from PIL import Image
 
-# ---- Core pipeline wrappers (from core/app.py) ----
-from core.app import (
-    load_models_with_validation,
-    process_video_fixed,
-    get_model_status,
-    get_cache_status,
-    PROCESS_CANCELLED,
-)
+# DO NOT import from core.app here!
+# We'll get these via parameter injection or lazy imports
 
 # ---- Optional utilities (background generator & previews) ----
 _try_bg_gen = None
@@ -134,6 +124,8 @@ def _blend(n, pal):
 # ------------------------------------------------------------------
 def cb_load_models() -> str:
     """Load SAM2 + MatAnyOne and return human-readable status."""
+    # Lazy import to avoid circular dependency
+    from core.app import load_models_with_validation
     return load_models_with_validation()
 
 
@@ -143,7 +135,7 @@ def cb_load_models() -> str:
 def cb_process_video(
     vid: str,
     style: str,
-    custom_bg_path: str | None,  # Now directly a filepath from gr.Image
+    custom_bg_path: str | None,
     use_two: bool,
     chroma: str,
     key_color_mode: str,
@@ -154,18 +146,18 @@ def cb_process_video(
     Runs the two-stage (or single-stage) pipeline and returns:
         (processed_video_path | None, status_message:str)
     """
+    # Lazy imports to avoid circular dependency
+    from core.app import process_video_fixed, PROCESS_CANCELLED
+    
     # Reset any prior cancel flag when user clicks Run
     if PROCESS_CANCELLED.is_set():
         PROCESS_CANCELLED.clear()
 
-    # custom_bg_path is now directly a filepath string from gr.Image
-    # No need to extract from dict
-
     # Fire the core function
     return process_video_fixed(
         video_path=vid,
         background_choice=style,
-        custom_background_path=custom_bg_path,  # Direct path
+        custom_background_path=custom_bg_path,
         progress_callback=None,
         use_two_stage=use_two,
         chroma_preset=chroma,
@@ -180,6 +172,7 @@ def cb_process_video(
 # ------------------------------------------------------------------
 def cb_cancel() -> str:
     try:
+        from core.app import PROCESS_CANCELLED
         PROCESS_CANCELLED.set()
         return "Cancellation requested."
     except Exception as e:
@@ -187,6 +180,7 @@ def cb_cancel() -> str:
 
 def cb_status() -> Tuple[Dict[str, Any], Dict[str, Any]]:
     try:
+        from core.app import get_model_status, get_cache_status
         return get_model_status(), get_cache_status()
     except Exception as e:
         return {"error": str(e)}, {"error": str(e)}