track tqdm in main function too

app.py

@@ -301,9 +301,9 @@ def infer(video_in, trim_value, prompt, background_prompt, progress=gr.Progress(
     print(prompt)
     break_vid = get_frames(video_in)
     
-    frames_list= break_vid[0]
+    frames_list = break_vid[0]
     fps = break_vid[1]
-    n_frame = int(trim_value*fps)
+    n_frame = int(trim_value * fps)
     
     if n_frame >= len(frames_list):
         print("video is shorter than the cut value")
@@ -313,36 +313,36 @@ def infer(video_in, trim_value, prompt, background_prompt, progress=gr.Progress(
     with_green_result_frames = []
     with_matte_result_frames = []
     
-    print("set stop frames to: " + str(n_frame))
+    print("set stop frames to:", n_frame)
     bg_already = False
-    for i in frames_list[0:int(n_frame)]:
+
+    # tqdm hooked into gr.Progress
+    for i in tqdm(frames_list[:n_frame], desc="Inferring frames", unit="frame"):
         to_numpy_i = Image.open(i).convert("RGB")
-        #need to convert to numpy
-        # Convert the image to a NumPy array
         image_array = np.array(to_numpy_i)
 
         results = run_grounded_sam(image_array, prompt, "text", background_prompt, bg_already)
         bg_already = True
+
         bg_img = Image.fromarray(results[0])
         green_img = Image.fromarray(results[1])
         matte_img = Image.fromarray(results[2])
         
-  
         # exporting the images
         bg_img.save(f"bg_result_img-{i}.jpg")
         with_bg_result_frames.append(f"bg_result_img-{i}.jpg")
+
         green_img.save(f"green_result_img-{i}.jpg")
         with_green_result_frames.append(f"green_result_img-{i}.jpg")
+
         matte_img.save(f"matte_result_img-{i}.jpg")
         with_matte_result_frames.append(f"matte_result_img-{i}.jpg")
-        print("frame " + i + "/" + str(n_frame) + ": done;")
 
     vid_bg = create_video(with_bg_result_frames, fps, "bg")
     vid_green = create_video(with_green_result_frames, fps, "greenscreen")
     vid_matte = create_video(with_matte_result_frames, fps, "matte")
 
-    bg_already = False
-    print("finished !")
+    print("finished!")
     
     return vid_bg, vid_green, vid_matte