Omnibus commited on
Commit
3582ba1
·
1 Parent(s): 1c81419

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -35,10 +35,15 @@ def load_video(vid):
35
  new_video_in = str(vid)
36
  capture = cv2.VideoCapture(new_video_in)
37
  fps = capture.get(cv2.CAP_PROP_FPS)
38
- #print (f'FPS:: {fps}')
39
  frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
40
  capture.release()
41
- return frame_count, fps
 
 
 
 
 
42
 
43
  def capture_function(vid=None):
44
  new_video_in = str(vid)
@@ -197,7 +202,7 @@ with gr.Blocks() as app:
197
  return inp
198
  #outp_vid.change(echo_fn,outp_vid,[out_trim])
199
  trim_btn.click(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid])
200
- outp_vid.change(load_video,outp_vid,[frame_count,fps]).then(echo_fn,outp_vid,[out_trim])
201
  frame_btn.click(capture_function,[out_trim],[frame_gal,frame_file])
202
  rem_btn.click(rem_bg,[out_trim],[rem_vid,frame_num,rem_bg_vid])
203
  go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim,hid_box])
 
35
  new_video_in = str(vid)
36
  capture = cv2.VideoCapture(new_video_in)
37
  fps = capture.get(cv2.CAP_PROP_FPS)
38
+ msc = cap.get(cv2.CAP_PROP_POS_MSEC)
39
  frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
40
  capture.release()
41
+ seconds = int(msc/1000)
42
+ minutes = int(msc/1000)/60
43
+ hours = int(msc/1000)/360
44
+ vid_len = f'{hours}:{minutes}:{seconds}'
45
+ vid_len_up = gr.update(label = "End", value = f"{vid_len}", placeholder = "0:00:54")
46
+ return frame_count, fps, vid_len_up
47
 
48
  def capture_function(vid=None):
49
  new_video_in = str(vid)
 
202
  return inp
203
  #outp_vid.change(echo_fn,outp_vid,[out_trim])
204
  trim_btn.click(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid])
205
+ outp_vid.change(load_video,outp_vid,[frame_count,fps,end_f]).then(echo_fn,outp_vid,[out_trim])
206
  frame_btn.click(capture_function,[out_trim],[frame_gal,frame_file])
207
  rem_btn.click(rem_bg,[out_trim],[rem_vid,frame_num,rem_bg_vid])
208
  go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim,hid_box])