Create ui/callbacks.py

ui/callbacks.py

@@ -0,0 +1,75 @@
+"""
+Callbacks for BackgroundFX Pro UI
+Pure functions wired to UI events.
+"""
+
+import os
+from typing import Any, Dict, Tuple
+
+from core.app import (
+    load_models_with_validation,
+    process_video_fixed,
+    get_model_status,
+    get_cache_status,
+    PROCESS_CANCELLED,
+)
+from utils.bg_generator import generate_ai_background
+from utils.media_utils import first_frame, pil_from_path
+
+# -------- MODEL MANAGEMENT --------
+def cb_load_models() -> str:
+    return load_models_with_validation()
+
+def cb_process_video(
+    vid: str, style: str, custom_file: dict | None,
+    use_two: bool, chroma: str, prev_mask: bool, prev_green: bool,
+):
+    if PROCESS_CANCELLED.is_set():
+        PROCESS_CANCELLED.clear()
+    custom_path = None
+    if isinstance(custom_file, dict) and custom_file.get("name"):
+        custom_path = custom_file["name"]
+    return process_video_fixed(
+        video_path=vid,
+        background_choice=style,
+        custom_background_path=custom_path,
+        progress_callback=None,
+        use_two_stage=use_two,
+        chroma_preset=chroma,
+        preview_mask=prev_mask,
+        preview_greenscreen=prev_green,
+    )
+
+def cb_cancel() -> str:
+    try:
+        PROCESS_CANCELLED.set()
+        return "Cancellation requested."
+    except Exception as e:
+        return f"Cancel failed: {e}"
+
+def cb_status() -> Tuple[Dict[str, Any], Dict[str, Any]]:
+    try:
+        return get_model_status(), get_cache_status()
+    except Exception as e:
+        return {"error": str(e)}, {"error": str(e)}
+
+def cb_clear():
+    return None, "", None, ""
+
+# -------- AI BACKGROUND --------
+def cb_generate_bg(prompt_text: str, w: int, h: int, b: float, v: float, c: float):
+    img, path = generate_ai_background(prompt_text, width=int(w), height=int(h), bokeh=b, vignette=v, contrast=c)
+    return img, path
+
+def cb_use_gen_bg(path_text: str):
+    return {"name": path_text, "size": os.path.getsize(path_text)} if path_text and os.path.exists(path_text) else None
+
+# -------- PREVIEWS --------
+def cb_video_changed(vid_path: str):
+    img, info = first_frame(vid_path)
+    return img  # Could output preview img if UI has a slot
+
+def cb_custom_bg_preview(file_obj: dict | None):
+    if isinstance(file_obj, dict) and file_obj.get("name") and os.path.exists(file_obj["name"]):
+        return pil_from_path(file_obj["name"])
+    return None