Update app.py

app.py

@@ -14,9 +14,22 @@ for model_id in model_ids:
     snapshot_download(model_id, local_dir=f'checkpoints/{model_name}')
 
 import subprocess
+from moviepy.editor import VideoFileClip
+
+def get_video_properties(filepath):
+    video = VideoFileClip(filepath)
+    duration = video.duration
+    fps = video.fps
+    video.close()
+    return duration, fps
+
+
+
 
 def run_inference(prompt, video_path, condition, video_length):
-    
+    # Call the function to get the video properties
+    duration, fps = get_video_properties(video_path)
+    video_length = video_length * fps
     output_path = 'output/'
     os.makedirs(output_path, exist_ok=True)
     command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --video_length {video_length} --smoother_steps 19 20"