Update app.py

app.py

@@ -6,8 +6,11 @@ import rembg
 import cv2
 import uuid
 uid=uuid.uuid4()
-if not os.path.exists(f'{uid}-frames'): os.makedirs(f'{uid}')
-if not os.path.exists(f'{uid}-rembg'): os.makedirs(f'{uid}')
+if not os.path.exists(f'{uid}-frames'): os.makedirs(f'{uid}-frames')
+if not os.path.exists(f'{uid}-rembg'): os.makedirs(f'{uid}-rembg')
+
+
+
 
 load_js = """
 
@@ -31,12 +34,49 @@ def rem_cv(inp):
         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)
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     tmp_frame='1'
     capture.release()
+
+
+
+    
     return frame_count