Drop_videos_here

@echo off

set "FACE_IMG="
set "VIDEOS="
set "OUTPUT_DIR=output\"

if "%~1"=="" (
    echo Pass at least one video file!
    pause
    goto :eof
)

echo Path to face image:
set /p "FACE_IMG="

for %%i in ("%FACE_IMG%") do (
    if "%%~xi"==".jpg" set "FACE_IMG=%%i"
    if "%%~xi"==".jpeg" set "FACE_IMG=%%i"
    if "%%~xi"==".png" set "FACE_IMG=%%i"
)

if not defined FACE_IMG goto wrong_args
echo Face image file is: %FACE_IMG%
for %%i in ("%FACE_IMG%") do set "FACE_NAME=%%~ni"

call %userprofile%\anaconda3\Scripts\activate.bat %userprofile%\anaconda3
cd %~dp0
call activate simswap
:swap_video
echo Swapping video: %~nx1
call python test_video_swapmutil.py --isTrain false --name people --Arc_path arcface_model/arcface_checkpoint.tar --temp_path ./temp_results ^
        --pic_a_path %FACE_IMG% ^
        --video_path %~1 ^
        --output_path "./%OUTPUT_DIR%%FACE_NAME%_%~n1.mp4"
)
shift
if not "%~1"=="" goto :swap_video
if exist %~dp0%OUTPUT_DIR% call explorer /select,%~dp0%OUTPUT_DIR%
goto :eof

:wrong_args
echo No face image provided^!
echo Args: %~1 %VIDEO%