add tqdm tracking

app.py

@@ -5,6 +5,7 @@ import os
 import sys
 import random
 import warnings
+from tqdm import tqdm
 
 
 os.system("export BUILD_WITH_CUDA=True")
@@ -74,39 +75,44 @@ generator.to(device)
 
 def get_frames(video_in):
     frames = []
-    #resize the video
+    # resize the video
     clip = VideoFileClip(video_in)
-    
-    #check fps
+
+    # check fps
     if clip.fps > 30:
-        print("vide rate is over 30, resetting to 30")
+        print("video rate is over 30, resetting to 30")
         clip_resized = clip.resize(height=512)
-        clip_resized.write_videofile("video_resized.mp4", fps=30)
+        clip_resized.write_videofile("video_resized.mp4", fps=30, verbose=False, logger=None)
     else:
         print("video rate is OK")
         clip_resized = clip.resize(height=512)
-        clip_resized.write_videofile("video_resized.mp4", fps=clip.fps)
-    
+        clip_resized.write_videofile("video_resized.mp4", fps=clip.fps, verbose=False, logger=None)
+
     print("video resized to 512 height")
-    
+
     # Opens the Video file with CV2
-    cap= cv2.VideoCapture("video_resized.mp4")
-    
+    cap = cv2.VideoCapture("video_resized.mp4")
+
     fps = cap.get(cv2.CAP_PROP_FPS)
-    print("video fps: " + str(fps))
-    i=0
-    while(cap.isOpened()):
-        ret, frame = cap.read()
-        if ret == False:
-            break
-        cv2.imwrite('kang'+str(i)+'.jpg',frame)
-        frames.append('kang'+str(i)+'.jpg')
-        i+=1
-    
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))  # total number of frames
+    print(f"video fps: {fps}, total frames: {total_frames}")
+
+    i = 0
+    with tqdm(total=total_frames, desc="Extracting frames", unit="frame") as pbar:
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+            frame_name = f'kang{i}.jpg'
+            cv2.imwrite(frame_name, frame)
+            frames.append(frame_name)
+            i += 1
+            pbar.update(1)
+
     cap.release()
     cv2.destroyAllWindows()
     print("broke the video into frames")
-    
+
     return frames, fps
 
 
@@ -291,7 +297,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, background_prompt, bg_
     #return [(com_img, 'composite with background'), (green_img, 'green screen'), (alpha_rgb, 'alpha matte')]
     return com_img, green_img, alpha_rgb
 
-def infer(video_in, trim_value, prompt, background_prompt):
+def infer(video_in, trim_value, prompt, background_prompt, progress=gr.Progress(track_tqdm=True)):
     print(prompt)
     break_vid = get_frames(video_in)