minor changes 2

app.py

@@ -74,18 +74,26 @@ def swap_video(src_img, src_idx, video, dst_idx, progress=gr.Progress(track_tqdm
         dst_video_path = video
 
     from VideoSwapping import extract_frames, frames_to_video
+
+    # Extract frames only if not already present
     frame_paths = extract_frames(dst_video_path, frames_dir)
     log += f"Extracted {len(frame_paths)} frames to {frames_dir}\n"
     progress(0.15, desc="Extracted frames")
 
+    # Prepare swapped frames list and resume if possible
+    swapped_files = set(os.listdir(swapped_dir))
     for idx, frame_path in enumerate(frame_paths):
-        out_path = os.path.join(swapped_dir, f"swapped_{idx:05d}.jpg")
+        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)})")
+            continue
         try:
             try:
                 swapped = swapper.swap_faces(src_path, int(src_idx), frame_path, int(dst_idx))
             except ValueError as ve:
                 if int(dst_idx) != 1 and "Target image contains" in str(ve):
-                    # Fallback to dst_idx=1 if requested index not found in this frame
                     swapped = swapper.swap_faces(src_path, int(src_idx), frame_path, 1)
                     log += f"Frame {idx}: dst_idx {dst_idx} not found, used 1 instead.\n"
                 else: