Update app.py

app.py

@@ -17,24 +17,23 @@ import subprocess
 
 import cv2
 
-def get_frame_count_in_duration(filepath, duration):
+def get_frame_count_in_duration(filepath):
     video = cv2.VideoCapture(filepath)
     fps = video.get(cv2.CAP_PROP_FPS)
-    frame_count = int(fps * duration)
+    frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
+    duration = frame_count / fps
+    width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
     video.release()
-    return frame_count
+    return gr.update(maximum=frame_count)
 
 
 
 def run_inference(prompt, video_path, condition, video_length):
-    # Call the function to get the video properties
-    video_length = get_frame_count_in_duration(video_path, video_length)
-    print(video_length)
-    nb_frame = int(video_length)
-    #video_length = int(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 {nb_frame} --smoother_steps 19 20"
+    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"
     subprocess.run(command, shell=True)
 
     # Construct the video path
@@ -55,7 +54,10 @@ with gr.Blocks() as demo:
         submit_btn = gr.Button("Submit")
         video_res = gr.Video(label="result")
         status = gr.Textbox(label="result")
-
+    video_path.change(fn=get_frame_count_in_duration,
+                      inputs=[video_path],
+                      outputs=[video_length]
+                     )
     submit_btn.click(fn=run_inference, 
                      inputs=[prompt,
                              video_path,