@echo off

set PYTHON_VER=3.10.9
::set CUDA_VISIBLE_DEVICES=1
:: Deactivate the virtual environment
call .\venv\Scripts\deactivate.bat

:: Activate the virtual environment
call .\venv\Scripts\activate.bat
set PATH=%PATH%;%~dp0venv\Lib\site-packages\torch\lib

:: If the first argument is --help, skip the validation step
if "%~1" equ "--help" goto :skip_validation

:: Validate requirements
::python.exe .\setup\validate_requirements.py
::if %errorlevel% neq 0 exit /b %errorlevel%

:skip_validation

:: If the exit code is 0, run the kohya_gui.py script with the command-line arguments
if %errorlevel% equ 0 (
    REM Check if the batch was started via double-click
    IF /i "%comspec% /c %~0 " equ "%cmdcmdline:"=%" (
        REM echo This script was started by double clicking.
        cmd /k python.exe kohya_gui.py %*
    ) ELSE (
        REM echo This script was started from a command prompt.
        python.exe kohya_gui.py %*
    )
)