corrected frame index order

app.py

@@ -105,6 +105,7 @@ def get_frames(video_in, img_type):
     clip = VideoFileClip(video_in)
     
     #check fps
+    
     if clip.fps > 30:
         print("vide rate is over 30, resetting to 30")
         clip_resized = clip.resize(height=512)
@@ -114,6 +115,7 @@ def get_frames(video_in, img_type):
         clip_resized = clip.resize(height=512)
         clip_resized.write_videofile(f"{img_type}_video_resized.mp4", fps=clip.fps)
     
+    
     print("video resized to 512 height")
     
     # Opens the Video file with CV2
@@ -127,11 +129,13 @@ def get_frames(video_in, img_type):
         if ret == False:
             break
         if img_type == "source":
-            cv2.imwrite('kang'+str(i)+'.jpg',frame)
-            frames.append('kang'+str(i)+'.jpg')
+            filename = f'{i:05d}.jpg'
+            cv2.imwrite(filename, frame)
+            frames.append(filename)
         elif img_type == "mask":
-            cv2.imwrite('kang'+str(i)+'.png',frame)
-            frames.append('kang'+str(i)+'.png')
+            filename = f'{i:05d}.png'
+            cv2.imwrite(filename, frame)
+            frames.append(filename)
         i+=1
     
     cap.release()
@@ -161,12 +165,10 @@ def infer_auto(project_name, video_in, subject_to_remove):
 
     video_frames = get_frames(video_in, "source")
     print(video_frames[0])
-    #video_frames_paths = [os.path.abspath(image) for image in video_frames[0]]
-
-   
+ 
     masks_frames = get_frames(matte_video, "mask")
     print(masks_frames[0])
-    #masks_frames_paths = [os.path.abspath(image) for image in masks_frames[0]]
+    
 
     # Create the directory if it doesn't exist
     my_video_directory = f"{project_name}"