Update app.py

app.py

@@ -222,17 +222,13 @@ def infer_auto(project_name, video_in, subject_to_remove):
 
 
     #bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
-    
-    # Convert the float fps to an integer
-    needed_fps = int(video_frames[1])
-    
+        
     command = [
       f"python", 
       f"inference_propainter.py",
       f"--video={video_frames_folder}",
       f"--mask={masks_folder}",
-      f"--output={output_folder}",
-      f"--save_fps={needed_fps}"
+      f"--output={output_folder}"
     ]
 
     execute_command(command)
@@ -253,7 +249,27 @@ def infer_auto(project_name, video_in, subject_to_remove):
     for item in results_folder_content:
         print(item)
 
-    return "done", results_folder_content[0], results_folder_content[1]
+
+    # Convert the float fps to an integer
+    needed_fps = int(video_frames[1])
+
+    # Load the videos
+    video1 = VideoFileClip(results_folder_content[0])
+    video2 = VideoFileClip(results_folder_content[1])
+
+    # Set the desired frame rate (e.g., 12 fps)
+    new_fps = needed_fps
+
+    # Convert videos to the new frame rate
+    video1 = video1.set_duration(video1.duration).set_fps(new_fps)
+    video2 = video2.set_duration(video2.duration).set_fps(new_fps)
+
+    # Save the converted videos
+    video1.write_videofile(f'masked_{new_fps}fps.mp4', fps=new_fps)
+    video2.write_videofile(f'final_{new_fps}fps.mp4', fps=new_fps)
+
+
+    return "done", f'masked_{new_fps}fps.mp4', f'final_{new_fps}fps.mp4'
 
 css="""
 #col-container{