Update app.py

app.py

@@ -175,10 +175,26 @@ def infer_auto(project_name, video_in, subject_to_remove):
     #video_cut = f"video_cut.mp4"
     #ffmpeg_extract_subclip(video_in, t1=0, t2=3, targetname=video_cut)
     video_frames = get_frames(video_in, "source")
+    frames_list = video_frames[0]
     print(video_frames[0])
  
     masks_frames = get_frames(matte_video, "mask")
+    masks_list = masks_frames[0]
     print(masks_frames[0])
+
+
+    # Check the lengths of the two lists
+    frames_length = len(frames_list)
+    masks_length = len(masks_list)
+
+    # Make the lists the same length if they are different
+    if frames_length > masks_length:
+        frames_list = frames_list[:masks_length]
+    elif masks_length > frames_length:
+        masks_list = masks_list[:frames_length]
+
+    # Now, both lists have the same length
+
     
 
     # Create the directory if it doesn't exist
@@ -194,7 +210,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
 
 
     # Assuming 'images' is a list of image file paths
-    for idx, image in enumerate(video_frames[0]):
+    for idx, image in enumerate(frames_list):
         # Get the base file name (without path) from the original location
         image_name = os.path.basename(image)
     
@@ -220,7 +236,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
     print(f"Created the dir: {my_masks_directory}")
 
     # Assuming 'images' is a list of image file paths
-    for idx, image in enumerate(masks_frames[0]):
+    for idx, image in enumerate(masks_list):
         # Get the base file name (without path) from the original location
         image_name = os.path.basename(image)