Update app.py

app.py

@@ -29,6 +29,13 @@ def get_frame_count_in_duration(filepath):
     video.release()
     return gr.update(maximum=frame_count)
 
+def get_video_dimension(filepath):
+    video = cv2.VideoCapture(filepath)
+    width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    video.release()
+    return width, height
+
 def resize_video(input_path, output_path, width):
     # Load the video clip
     video = VideoFileClip(input_path)
@@ -52,6 +59,9 @@ def run_inference(prompt, video_path, condition, video_length):
 
     # Call the function to resize the video
     video_path = resize_video(input_vid, resized_vid, width=512)
+    width, height = get_video_dimension(video_path)
+
+    print(f"{width} x {height}")
     
     output_path = 'output/'
     os.makedirs(output_path, exist_ok=True)
@@ -65,9 +75,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}' --video_length {video_length} --is_long_video"
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width {width} --height {height} --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}' --video_length {video_length}"
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width {width} --height {height} --video_length {video_length}"
     subprocess.run(command, shell=True)
 
     # Construct the video path