Update ui/callbacks.py

ui/callbacks.py

@@ -8,7 +8,7 @@
 • PREVIEW FUNCTIONS NOW IMPLEMENTED:
     - cb_video_changed        → returns first frame
     - cb_custom_bg_preview    → returns uploaded image
-• UPDATED: Background preview persists during processing
+• FIXED: Returns only 2 values for process_video (don't touch preview)
 """
 
 from __future__ import annotations
@@ -33,9 +33,6 @@
 except Exception:
     pass
 
-# Global to store current background for persistence
-current_background_image = None
-
 
 # ------------------------------------------------------------------
 # LIGHTWEIGHT BG GENERATOR (inline fallback)
@@ -156,11 +153,9 @@ def cb_process_video(
 ):
     """
     Runs the two-stage (or single-stage) pipeline and returns:
-        (processed_video_path | None, status_message:str, background_preview)
-    Now returns THREE values to maintain background preview during processing.
+        (processed_video_path | None, status_message:str)
+    Returns ONLY 2 values - doesn't touch the background preview.
     """
-    global current_background_image
-    
     # Reset any prior cancel flag when user clicks Run
     if PROCESS_CANCELLED.is_set():
         PROCESS_CANCELLED.clear()
@@ -169,23 +164,9 @@ def cb_process_video(
     custom_path = None
     if isinstance(custom_file, dict) and custom_file.get("name"):
         custom_path = custom_file["name"]
-        # Update the background preview for custom uploads
-        try:
-            img = Image.open(custom_path).convert("RGB")
-            current_background_image = np.array(img)
-        except Exception:
-            pass
-    elif not custom_file and style:
-        # Generate preview for preset background
-        try:
-            from utils.cv_processing import create_professional_background
-            # Create a preview-sized version of the preset
-            current_background_image = create_professional_background(style, 640, 360)
-        except Exception:
-            pass
 
     # Fire the core function
-    video_path, status = process_video_fixed(
+    return process_video_fixed(
         video_path=vid,
         background_choice=style,
         custom_background_path=custom_path,
@@ -196,9 +177,6 @@ def cb_process_video(
         preview_mask=prev_mask,
         preview_greenscreen=prev_green,
     )
-    
-    # Return video, status, AND the background preview to keep it visible
-    return video_path, status, current_background_image
 
 
 # ------------------------------------------------------------------
@@ -219,8 +197,6 @@ def cb_status() -> Tuple[Dict[str, Any], Dict[str, Any]]:
 
 def cb_clear():
     """Clear all outputs including the background preview"""
-    global current_background_image
-    current_background_image = None
     # Return blanks for (out_video, status, gen_preview, gen_path, bg_preview)
     return None, "", None, "", None
 
@@ -229,23 +205,13 @@ def cb_clear():
 # AI BACKGROUND
 # ------------------------------------------------------------------
 def cb_generate_bg(prompt_text: str, w: int, h: int, b: float, v: float, c: float):
-    """Generate AI background and update global preview"""
-    global current_background_image
+    """Generate AI background"""
     img, path = _generate_ai_background(prompt_text, int(w), int(h), b, v, c)
-    # Store as numpy array for consistency
-    if img:
-        current_background_image = np.array(img)
     return img, path
 
 def cb_use_gen_bg(path_text: str):
-    """Use generated background as custom and update preview"""
-    global current_background_image
+    """Use generated background as custom"""
     if path_text and os.path.exists(path_text):
-        try:
-            img = Image.open(path_text).convert("RGB")
-            current_background_image = np.array(img)
-        except Exception:
-            pass
         return {"name": path_text, "size": os.path.getsize(path_text)}
     return None
 
@@ -275,14 +241,11 @@ def cb_video_changed(vid_path: str):
 def cb_custom_bg_preview(file_obj: dict | None):
     """
     Display the uploaded background image (if any) as a preview.
-    Stores it globally so it persists during processing.
     """
-    global current_background_image
     try:
         if file_obj and file_obj.get("name"):
             img = Image.open(file_obj["name"]).convert("RGB")
-            current_background_image = np.array(img)
-            return current_background_image
+            return np.array(img)
     except Exception:
         pass
     return None
@@ -292,11 +255,10 @@ def cb_preset_bg_preview(style: str):
     Generate and display preview for preset backgrounds.
     Called when background style dropdown changes.
     """
-    global current_background_image
     try:
         from utils.cv_processing import create_professional_background
         # Create a preview-sized version
-        current_background_image = create_professional_background(style, 640, 360)
-        return current_background_image
+        preview_bg = create_professional_background(style, 640, 360)
+        return preview_bg
     except Exception:
         return None