Update app.py

app.py

@@ -29,10 +29,29 @@ def get_frame_count_in_duration(filepath):
     video.release()
     return gr.update(maximum=frame_count)
 
+def resize_video(input_path, output_path, width):
+    # Load the video clip
+    video = VideoFileClip(input_path)
 
+    # Calculate the new height while maintaining the aspect ratio
+    height = int(video.size[1] * (width / video.size[0]))
+
+    # Resize the video
+    resized_video = video.resize(width=width, height=height)
+
+    # Write the resized video to the output path
+    resized_video.write_videofile(output_path, codec='libx264')
+
+    return output_path
         
 def run_inference(prompt, video_path, condition, video_length):
 
+    # Specify the input and output paths
+    input_vid = video_path
+    resized_vid = 'resized.mp4'
+
+    # Call the function to resize the video
+    video_path = resize_video(input_vid, resized_vid, width=512)
     
     output_path = 'output/'
     os.makedirs(output_path, exist_ok=True)