Update app.py

app.py

@@ -37,38 +37,22 @@ def get_video_dimension(filepath):
     video.release()
     return width, height, fps
 
-def get_new_dimensions(width, height):
-    new_width = width
-    new_height = height
-
-    # Adjust width to be a multiple of 32
-    while new_width % 32 != 0:
-        new_width += 1
-
-    # Adjust height to maintain the aspect ratio
-    new_height = int(new_width * (height / new_width))
-    
-
-    # Adjust height to be a multiple of 32
-    while new_height % 32 != 0:
-        new_height += 1
-
-    return new_width, new_height
-
 def resize_video(input_path, output_path, target_width):
     # Load the video clip
-    video = VideoFileClip(input_path)
+    clip = VideoFileClip(input_path)
 
-    # Calculate the new dimensions while maintaining the aspect ratio and multiples of 32
-    new_width, new_height = get_new_dimensions(target_width, video.size[1])
+    # Calculate the new dimensions while maintaining the aspect ratio
+    target_height = int(clip.size[1] * target_width / clip.size[0])
+    target_height = target_height + (32 - target_height % 32) if target_height % 32 != 0 else target_height
 
-    # Resize the video
-    resized_video = video.resize(width=new_width, height=new_height)
+    # Resize the video clip
+    resized_clip = clip.resize(width=target_width, height=target_height)
 
-    # Write the resized video to the output path
-    resized_video.write_videofile(output_path, codec='libx264')
+    # Write the resized video to the output file
+    resized_clip.write_videofile(output_path, codec='libx264')
 
-    return output_path
+    # Close the video clip
+    clip.close()
         
 def run_inference(prompt, video_path, condition, video_length):
 
@@ -77,8 +61,8 @@ def run_inference(prompt, video_path, condition, video_length):
     resized_vid = 'resized.mp4'
 
     # Call the function to resize the video
-    video_path = resize_video(input_vid, resized_vid, target_width=512)
-    width, height, fps = get_video_dimension(video_path)
+    resized_video_path = resize_video(input_vid, resized_vid, target_width=512)
+    width, height, fps = get_video_dimension(resized_video_path)
 
     print(f"{width} x {height} | {fps}")
     
@@ -94,9 +78,9 @@ def run_inference(prompt, video_path, condition, video_length):
         os.remove(video_path_output)
 
     if video_length > 12:
-        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width {width} --height {height} --fps {fps} --video_length {video_length} --is_long_video"
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{resized_video_path}' --output_path '{output_path}' --width {width} --height {height} --fps {fps} --video_length {video_length} --is_long_video"
     else:
-        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width {width} --height {height} --fps {fps} --video_length {video_length}"
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{resized_video_path}' --output_path '{output_path}' --width {width} --height {height} --fps {fps} --video_length {video_length}"
     subprocess.run(command, shell=True)
 
     # Construct the video path