Update app.py

app.py

@@ -15,14 +15,21 @@ for model_id in model_ids:
 
 import subprocess
 
+import cv2
 
+def get_frame_count_in_duration(filepath, duration):
+    video = cv2.VideoCapture(filepath)
+    fps = video.get(cv2.CAP_PROP_FPS)
+    frame_count = int(fps * duration)
+    video.release()
+    return frame_count
 
 
 
 def run_inference(prompt, video_path, condition, video_length):
     # Call the function to get the video properties
-    # duration, fps = get_video_properties(video_path)
-    #print(fps)
+    video_length = get_video_properties(video_path, video_length)
+    print(video_length)
     #video_length = int(video_length * fps)
     output_path = 'output/'
     os.makedirs(output_path, exist_ok=True)