From 1c03fe3f1b4478e03a1ef00d7db20d9c39f347d7 Mon Sep 17 00:00:00 2001
From: Xavi Vinaixa <xavi@broomx.com>
Date: Mon, 5 Jun 2023 09:18:23 +0200
Subject: [PATCH] Replaced ProgressBar with the standard tqdm library for
 enhanced user experience #3

---
 refacer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/refacer.py b/refacer.py
index 686fc57..47bee10 100644
--- a/refacer.py
+++ b/refacer.py
@@ -58,6 +58,7 @@ class Refacer:
 
         cap = cv2.VideoCapture(video_path)
         total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+        print(f"Total frames: {total_frames}")
         
         fps = cap.get(cv2.CAP_PROP_FPS)
         frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
@@ -70,7 +71,7 @@ class Refacer:
         while cap.isOpened():
             flag, frame = cap.read()
             if flag and len(frame)>0:
-                pos_frame = cap.get(cv2.CAP_PROP_POS_FRAMES)
+                pos_frame = int(cap.get(cv2.CAP_PROP_POS_FRAMES))
                 
                 pbar.update(pos_frame)