Update processing/two_stage/two_stage_processor.py

processing/two_stage/two_stage_processor.py

@@ -446,17 +446,20 @@ def composite_frame(get_frame, t):
                     # Add channel dimension
                     alpha_frame = alpha_frame[:, :, np.newaxis]
                     
-                    # Prepare background
-                    bg_normalized = background.astype(np.float32) / 255.0
+                    # CRITICAL FIX: Ensure all inputs are in same scale (0-1)
+                    # MoviePy gives frames in 0-1 range, so normalize everything to 0-1
+                    original_normalized = original_frame.astype(np.float32)  # Already 0-1 from MoviePy
+                    bg_normalized = background.astype(np.float32) / 255.0    # Convert to 0-1
                     
-                    # DEBUG: Check background
+                    # DEBUG: Check all inputs are same scale
                     if t < 0.1:
-                        logger.info(f"🔍 DEBUG: Background shape: {bg_normalized.shape}, range: {bg_normalized.min():.3f} to {bg_normalized.max():.3f}")
+                        logger.info(f"🔍 DEBUG: Original normalized range: {original_normalized.min():.3f} to {original_normalized.max():.3f}")
+                        logger.info(f"🔍 DEBUG: Background normalized range: {bg_normalized.min():.3f} to {bg_normalized.max():.3f}")
                     
                     # Composite: person = alpha * original + (1-alpha) * background
                     # If alpha = 1 (white), show original person
                     # If alpha = 0 (black), show background
-                    result = alpha_frame * original_frame + (1 - alpha_frame) * bg_normalized
+                    result = alpha_frame * original_normalized + (1 - alpha_frame) * bg_normalized
                     
                     # DEBUG: Check result
                     if t < 0.1: