added time remaining

app.py

@@ -4,6 +4,7 @@ import cv2
 import numpy as np
 import shutil
 import subprocess
+import time
 from SinglePhoto import FaceSwapper
 
 wellcomingMessage = """
@@ -82,12 +83,18 @@ def swap_video(src_img, src_idx, video, dst_idx, progress=gr.Progress(track_tqdm
 
     # Prepare swapped frames list and resume if possible
     swapped_files = set(os.listdir(swapped_dir))
+    start_time = time.time()
+    total_frames = len(frame_paths)
     for idx, frame_path in enumerate(frame_paths):
         swapped_name = f"swapped_{idx:05d}.jpg"
         out_path = os.path.join(swapped_dir, swapped_name)
         if swapped_name in swapped_files and os.path.exists(out_path):
             log += f"Frame {idx}: already swapped, skipping.\n"
-            progress(0.15 + 0.6 * (idx + 1) / len(frame_paths), desc=f"Swapping frames ({idx+1}/{len(frame_paths)})")
+            elapsed = time.time() - start_time
+            avg_time = elapsed / (idx + 1) if idx + 1 > 0 else 0
+            remaining = avg_time * (total_frames - (idx + 1))
+            mins, secs = divmod(int(remaining), 60)
+            progress(0.15 + 0.6 * (idx + 1) / total_frames, desc=f"Swapping frames ({idx+1}/{total_frames}) | Est. left: {mins:02d}:{secs:02d}")
             continue
         try:
             try:
@@ -103,7 +110,11 @@ def swap_video(src_img, src_idx, video, dst_idx, progress=gr.Progress(track_tqdm
         except Exception as e:
             cv2.imwrite(out_path, cv2.imread(frame_path))
             log += f"Failed to swap frame {idx}: {e}\n"
-        progress(0.15 + 0.6 * (idx + 1) / len(frame_paths), desc=f"Swapping frames ({idx+1}/{len(frame_paths)})")
+        elapsed = time.time() - start_time
+        avg_time = elapsed / (idx + 1) if idx + 1 > 0 else 0
+        remaining = avg_time * (total_frames - (idx + 1))
+        mins, secs = divmod(int(remaining), 60)
+        progress(0.15 + 0.6 * (idx + 1) / total_frames, desc=f"Swapping frames ({idx+1}/{total_frames}) | Est. left: {mins:02d}:{secs:02d}")
     cap = cv2.VideoCapture(dst_video_path)
     fps = cap.get(cv2.CAP_PROP_FPS)
     cap.release()