Update app.py

app.py

@@ -43,13 +43,9 @@ def fn(vid, fps, color):
     processed_frames_changed_bg = []
     for frame in frames:
         pil_image = Image.fromarray(frame)
-        processed_image, mask = process(pil_image, color)  # Get both processed image and mask
-        processed_frames_no_bg.append(np.array(processed_image))  # Save no-background frame
-        
-        # Compose with background for changed background video
-        background = Image.new("RGBA", pil_image.size, color + str((255,)))
-        composed_image = Image.composite(pil_image, background, mask)
-        processed_frames_changed_bg.append(np.array(composed_image))
+        processed_image, mask = process(pil_image, color)  
+        processed_frames_no_bg.append(np.array(mask))  
+        processed_frames_changed_bg.append(np.array(processed_image))
 
     # Create a new video from the processed frames
     processed_video = mp.ImageSequenceClip(processed_frames_changed_bg, fps=fps)