Update app.py

app.py

@@ -29,63 +29,23 @@ def rem_cv(inp):
     
     while True:
         ret, frame = cap.read()
-        
         fgmask = fgbg.apply(frame)
         fgmask = cv2.morphologyEx(fgmask, cv2.MORPH_OPEN, kernel)
         yield (fgmask)
 
-def get_rate(vid_in):
-    cap = cv2.VideoCapture(vid_in)
-
-    while(cap.isOpened()): 
-      
-        # Capture frame-by-frame 
-      
-        ret, frame = cap.read() 
-      
-        # if video finished or no Video Input 
-        if not ret: 
-            break
-      
-        # Our operations on the frame come here 
-        gray = frame 
-      
-        # resizing the frame size according to our need 
-        gray = cv2.resize(gray, (500, 300)) 
-      
-        # font which we will be using to display FPS 
-        font = cv2.FONT_HERSHEY_SIMPLEX 
-        # time when we finish processing for this frame 
-        new_frame_time = time.time() 
-      
-        # Calculating the fps 
-      
-        # fps will be number of frame processed in given time frame 
-        # since their will be most of time error of 0.001 second 
-        # we will be subtracting it to get more accurate result 
-        fps = 1/(new_frame_time-prev_frame_time) 
-        prev_frame_time = new_frame_time 
-
-
 def load_video(vid):
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)
     fps = capture.get(cv2.CAP_PROP_FPS)
-    print (f'FPS:: {fps}')
+    #print (f'FPS:: {fps}')
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     capture.release()
+    return frame_count, fps
 
-
-
-    
-    return frame_count 
-
-
-
-def capture_function(vid=None,frame_count=None):
+def capture_function(vid=None):
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)
-    #frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
+    frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     
     fbox=[]
     cnt=0
@@ -97,18 +57,14 @@ def capture_function(vid=None,frame_count=None):
         cv2.imwrite(f'{uid}/{i}.png',frame_ff)
         fbox.append(f'{uid}/{i}.png')
         
-    #capture.set(cv2.CAP_PROP_POS_FRAMES, int(frame_count)-1)
-    #ret, frame_f = capture.read()
-    #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
-    #cv2.imwrite(f'{uid}/{int(frame_count)-1}.png',frame_ff)
-    #fbox.append(f'{uid}/{int(frame_count)-1}.png')
-    #capture.release()
     return fbox,fbox
 
     
-def rem_bg(vid,frame_count):
+def rem_bg(vid):
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)    
+    frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
+    
     fbox2=[]
     cnt=0
     frame_count1= int(frame_count)
@@ -121,16 +77,7 @@ def rem_bg(vid,frame_count):
         cv2.imwrite(f'{uid}-rembg/{i}.png',out)
         fbox2.append(f'{uid}-rembg/{i}.png')
         yield fbox2
-        #capture.set(cv2.CAP_PROP_POS_FRAMES, int(frame_count)-1)
-        #ret, frame_f = capture.read()
-        #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
-        #cv2.imwrite(f'{uid}/{int(frame_count)-1}.png',frame_ff)
-        #fbox.append(f'{uid}/{int(frame_count)-1}.png')
-        #capture.release()
-        #out = cv2.VideoWriter('video.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)
-        #return fbox2
-
-    
+   
 
 
     
@@ -172,13 +119,16 @@ def dl(inp):
 
 with gr.Blocks() as app:
     with gr.Tab("Load"):
+        
         inp_url = gr.Textbox()
         go_btn = gr.Button("Run")
         with gr.Row():
             with gr.Column():
                 outp_vid=gr.Video()
             with gr.Column():
-                frame_count=gr.Textbox()
+                with gr.Row():
+                    frame_count=gr.Textbox(label="Frame Count",interactive = False)
+                    fps=gr.Textbox(label="FPS",interactive = False)
                 outp_file=gr.Files()
     with gr.Tab("Frames"):
         frame_btn = gr.Button("Get Frames")
@@ -200,7 +150,7 @@ with gr.Blocks() as app:
         text_input=gr.Textbox()
         text_output=gr.Textbox()
         url_params=gr.JSON()
-    outp_vid.change(load_video,outp_vid,frame_count)
+    outp_vid.change(load_video,outp_vid,[frame_count,fps])
     frame_btn.click(capture_function,[outp_vid,frame_count],[frame_gal,frame_file])
     rem_btn.click(rem_bg,[outp_vid,frame_count],rem_vid)
     go_btn.click(dl,inp_url,[outp_vid,in_vid,outp_file])