Update app.py

app.py

@@ -137,31 +137,33 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
         # Calculate the new height while maintaining the aspect ratio
         n_height = int(o_height / o_width * 512)
         n_width = 512
-    
-    # Get FPS of original video input
-    target_fps = get_video_dimension(video_path)[2] 
-    if target_fps > 12 :
-        print(f"FPS is too high: {target_fps}")
-        video_length = int((12 * video_length) / target_fps)
-            
-        target_fps = 12
-    print(f"NEW INPUT FPS: {target_fps}, NEW FPS: {video_length}")
-    
-    # Count total frames according to fps
-    total_frames = get_video_dimension(video_path)[3]
 
-    # Resize the video
+    # Make sure new dimensions are multipe of 32
     r_width = make_nearest_multiple_of_32(n_width)
     r_height = make_nearest_multiple_of_32(n_height)
     print(f"multiple of 32 sizes : {r_width}x{r_height}")
-    # Check if the file already exists
+
+    # Get FPS of original video input
+    original_fps = get_video_dimension(video_path)[2] 
+    if target_fps > 12 :
+        print(f"FPS is too high: {original_fps}")          
+        target_fps = 12
+    print(f"NEW INPUT FPS: {target_fps}, NEW LENGTH: {video_length}")
+    
+    # Check if the resized file already exists
     if os.path.exists('resized.mp4'):
         # Delete the existing file
         os.remove('resized.mp4')
+    
     resized = resize_video(video_path, 'resized.mp4', r_width, r_height, target_fps)
     resized_video_fcount = get_video_dimension(resized)[3]
     print(f"RESIZED VIDEO FRAME COUNT: {resized_video_fcount}")
 
+    # Make sure new total frame count is enough to handle chosen video length
+    # if video_length > resized_video_fcount :
+    # video_length = int((target_fps * video_length) / original_fps)
+    
+
     # normalize pixels
     #normalized = normalize_and_save_video(resized, 'normalized.mp4')