Update ui/callbacks.py

ui/callbacks.py

@@ -7,8 +7,7 @@
 • key_color_mode parameter already added (matches ui_components.py)
 • PREVIEW FUNCTIONS NOW IMPLEMENTED:
     - cb_video_changed        → returns first frame
-    - cb_custom_bg_preview    → returns uploaded image
-• FIXED: Returns only 2 values for process_video (don't touch preview)
+• SIMPLIFIED: Background handled by gr.Image component directly
 """
 
 from __future__ import annotations
@@ -144,7 +143,7 @@ def cb_load_models() -> str:
 def cb_process_video(
     vid: str,
     style: str,
-    custom_file: dict | None,
+    custom_bg_path: str | None,  # Now directly a filepath from gr.Image
     use_two: bool,
     chroma: str,
     key_color_mode: str,
@@ -154,22 +153,19 @@ def cb_process_video(
     """
     Runs the two-stage (or single-stage) pipeline and returns:
         (processed_video_path | None, status_message:str)
-    Returns ONLY 2 values - doesn't touch the background preview.
     """
     # Reset any prior cancel flag when user clicks Run
     if PROCESS_CANCELLED.is_set():
         PROCESS_CANCELLED.clear()
 
-    # Resolve custom background path (if provided)
-    custom_path = None
-    if isinstance(custom_file, dict) and custom_file.get("name"):
-        custom_path = custom_file["name"]
+    # 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_path,
+        custom_background_path=custom_bg_path,  # Direct path
         progress_callback=None,
         use_two_stage=use_two,
         chroma_preset=chroma,
@@ -196,8 +192,8 @@ def cb_status() -> Tuple[Dict[str, Any], Dict[str, Any]]:
         return {"error": str(e)}, {"error": str(e)}
 
 def cb_clear():
-    """Clear all outputs including the background preview"""
-    # Return blanks for (out_video, status, gen_preview, gen_path, bg_preview)
+    """Clear all outputs"""
+    # Return blanks for (out_video, status, gen_preview, gen_path, custom_bg)
     return None, "", None, "", None
 
 
@@ -209,10 +205,13 @@ def cb_generate_bg(prompt_text: str, w: int, h: int, b: float, v: float, c: floa
     img, path = _generate_ai_background(prompt_text, int(w), int(h), b, v, c)
     return img, path
 
-def cb_use_gen_bg(path_text: str):
-    """Use generated background as custom"""
-    if path_text and os.path.exists(path_text):
-        return {"name": path_text, "size": os.path.getsize(path_text)}
+def cb_use_gen_bg(gen_path: str):
+    """
+    Use generated background as custom.
+    Returns the path for gr.Image to display.
+    """
+    if gen_path and os.path.exists(gen_path):
+        return gen_path  # gr.Image can display from path
     return None
 
 
@@ -238,22 +237,10 @@ def cb_video_changed(vid_path: str):
     except Exception:
         return None
 
-def cb_custom_bg_preview(file_obj: dict | None):
-    """
-    Display the uploaded background image (if any) as a preview.
-    """
-    try:
-        if file_obj and file_obj.get("name"):
-            img = Image.open(file_obj["name"]).convert("RGB")
-            return np.array(img)
-    except Exception:
-        pass
-    return None
-
 def cb_preset_bg_preview(style: str):
     """
     Generate and display preview for preset backgrounds.
-    Called when background style dropdown changes.
+    Returns image for gr.Image component to display.
     """
     try:
         from utils.cv_processing import create_professional_background