Update app.py

app.py

@@ -44,15 +44,15 @@ def adjust_to_multiple_of_12(number):
         number += adjustment
     return number
     
-def resize_video(input_file, new_width):
+def resize_video(input_file):
     # Load the video clip
     clip = VideoFileClip(input_file)
     print(f"WIDTH TARGET: {new_width}")
     # Calculate the aspect ratio
-    ratio = new_width / clip.size[0]
+    ratio = 512 / clip.size[0]
     new_height = int(clip.size[1] * ratio)
     new_height_adjusted = adjust_to_multiple_of_12(new_height)
-    new_width_adjusted = adjust_to_multiple_of_12(new_width)
+    new_width_adjusted = adjust_to_multiple_of_12(512)
     print(f"OLD H: {new_height} | NEW H: {new_height_adjusted}")
     print(f"OLD W: {new_width} | NEW W: {new_width_adjusted}")
 
@@ -82,7 +82,7 @@ def run_inference(prompt, video_path, condition, video_length):
     input_vid = video_path
 
     # Call the function to resize the video
-    resized_video_path = resize_video(input_vid, new_width=512)
+    resized_video_path = resize_video(input_vid)
     print(f"PATH TO RESIZED: {resized_video_path}")
     width, height, fps = get_video_dimension(resized_video_path)
 
@@ -177,7 +177,7 @@ with gr.Blocks(css=css) as demo:
     video_path.change(fn=get_frame_count_in_duration,
                       inputs=[video_path],
                       outputs=[video_length]
-                     )
+                     ).then(fn=resize_video, inputs=[video_path], outputs=[video_path])
     submit_btn.click(fn=run_inference, 
                      inputs=[prompt,
                              video_path,