added copy to gdrive

app.py

@@ -341,7 +341,7 @@ def add_audio_to_video(original_video_path, video_no_audio_path, output_path):
     except subprocess.CalledProcessError as e:
         return False, e.stderr.decode()
 
-def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, add_audio=True, progress=gr.Progress()):
+def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, add_audio=True, copy_to_drive=False, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_path = "VideoSwapping/data_src.jpg"
@@ -413,6 +413,15 @@ def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, add_aud
     log += f"Combined swapped frames into video {output_video_path}\n"
     progress(0.8, desc="Muxing audio")
 
+    # Copy to Google Drive if requested
+    if copy_to_drive:
+        drive_path = "/content/drive/MyDrive/" + os.path.basename(output_video_path)
+        try:
+            shutil.copy(output_video_path, drive_path)
+            log += f"Copied swapped video without audio to Google Drive: {drive_path}\n"
+        except Exception as e:
+            log += f"Failed to copy to Google Drive: {e}\n"
+
     if add_audio:
         ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
         if ok:
@@ -444,7 +453,7 @@ def swap_video(src_img, src_idx, video, dst_idx, delete_frames_dir=True, add_aud
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return final_output_path, log
 
-def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=True, add_audio=True, progress=gr.Progress()):
+def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=True, add_audio=True, copy_to_drive=False, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_path = "VideoSwappingAllFaces/data_src.jpg"
@@ -524,6 +533,15 @@ def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=Tr
     log += f"Combined swapped frames into video {output_video_path}\n"
     progress(0.8, desc="Muxing audio")
 
+    # Copy to Google Drive if requested
+    if copy_to_drive:
+        drive_path = "/content/drive/MyDrive/" + os.path.basename(output_video_path)
+        try:
+            shutil.copy(output_video_path, drive_path)
+            log += f"Copied swapped video without audio to Google Drive: {drive_path}\n"
+        except Exception as e:
+            log += f"Failed to copy to Google Drive: {e}\n"
+
     if add_audio:
         ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
         if ok:
@@ -555,7 +573,7 @@ def swap_video_all_faces(src_img, video, num_faces_to_swap, delete_frames_dir=Tr
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return final_output_path, log
 
-def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=True, add_audio=True, progress=gr.Progress()):
+def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=True, add_audio=True, copy_to_drive=False, progress=gr.Progress()):
     log = ""
     start_time = time.time()
     src_dir = "CustomVideoSwap/src"
@@ -693,7 +711,7 @@ def swap_video_custom_mapping(src_imgs, video, mapping_str, delete_frames_dir=Tr
     log += f"Elapsed time: {elapsed:.2f} seconds\n"
     return final_output_path, log
 
-def swap_single_src_multi_video(src_img, dst_videos, dst_indices, delete_frames_dir=True, add_audio=True, progress=gr.Progress(track_tqdm=True)):
+def swap_single_src_multi_video(src_img, dst_videos, dst_indices, delete_frames_dir=True, add_audio=True, copy_to_drive=False, progress=gr.Progress(track_tqdm=True)):
     """
     Swaps a single source image onto multiple videos, with a mapping for destination face indices.
     Each video is processed one by one.
@@ -775,6 +793,15 @@ def swap_single_src_multi_video(src_img, dst_videos, dst_indices, delete_frames_
         frames_to_video(swapped_dir, output_video_path, fps)
         log += f"Combined swapped frames into video {output_video_path}\n"
 
+        # Copy to Google Drive if requested
+        if copy_to_drive:
+            drive_path = "/content/drive/MyDrive/" + os.path.basename(output_video_path)
+            try:
+                shutil.copy(output_video_path, drive_path)
+                log += f"Copied swapped video without audio to Google Drive: {drive_path}\n"
+            except Exception as e:
+                log += f"Failed to copy to Google Drive: {e}\n"
+
         # Mux audio if requested
         if add_audio:
             ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, output_video_with_audio)
@@ -885,7 +912,8 @@ with gr.Blocks() as demo:
                 gr.Video(label="Target Video"),
                 gr.Number(value=1, label="Destination Face Index"),
                 gr.Checkbox(label="Delete video_frames directory after processing", value=True),
-                gr.Checkbox(label="Add audio from original video", value=True)
+                gr.Checkbox(label="Add audio from original video", value=True),
+                gr.Checkbox(label="Copy swapped video (no audio) to Google Drive root", value=False)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -900,7 +928,8 @@ with gr.Blocks() as demo:
                 gr.Video(label="Target Video"),
                 gr.Number(value=1, label="Number of Faces to Swap"),
                 gr.Checkbox(label="Delete video_frames directory after processing", value=True),
-                gr.Checkbox(label="Add audio from original video", value=True)
+                gr.Checkbox(label="Add audio from original video", value=True),
+                gr.Checkbox(label="Copy swapped video (no audio) to Google Drive root", value=False)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -915,7 +944,8 @@ with gr.Blocks() as demo:
                 gr.Video(label="Target Video"),
                 gr.Textbox(label="Mapping (comma-separated, e.g. 2,1,3)"),
                 gr.Checkbox(label="Delete video_frames directory after processing", value=True),
-                gr.Checkbox(label="Add audio from original video", value=True)
+                gr.Checkbox(label="Add audio from original video", value=True),
+                gr.Checkbox(label="Copy swapped video (no audio) to Google Drive root", value=False)
             ],
             outputs=[
                 gr.Video(label="Swapped Video"),
@@ -930,7 +960,8 @@ with gr.Blocks() as demo:
                 gr.File(label="Target Videos (select multiple)", file_count="multiple", type="filepath"),
                 gr.Textbox(label="Destination Face Indices (comma-separated, e.g. 1,2,1)"),
                 gr.Checkbox(label="Delete video_frames directory after processing", value=True),
-                gr.Checkbox(label="Add audio from original video", value=True)
+                gr.Checkbox(label="Add audio from original video", value=True),
+                gr.Checkbox(label="Copy swapped video (no audio) to Google Drive root", value=False)
             ],
             outputs=[
                 gr.Gallery(label="Swapped Videos", type="filepath"),