setup.bat

@echo off
echo InferNode Platform Setup for Windows
echo ====================================
echo.

REM Check if Python is installed
python --version >nul 2>&1
if %errorlevel% neq 0 (
    echo Error: Python is not installed or not in PATH
    echo Please install Python 3.10-3.13 from https://python.org
    pause
    exit /b 1
)

REM Display Python version
echo Python detected:
python --version
echo.

REM Check Python version compatibility (requires 3.10 <= version <= 3.13 for Geti)
echo Checking Python version compatibility...
for /f "tokens=2" %%i in ('python --version 2^>^&1') do set PYTHON_VERSION=%%i
for /f "tokens=1,2 delims=." %%a in ("%PYTHON_VERSION%") do (
    set PYTHON_MAJOR=%%a
    set PYTHON_MINOR=%%b
)

if %PYTHON_MAJOR% LSS 3 (
    echo Error: Python 3.10-3.13 is required for Geti compatibility
    echo Found Python %PYTHON_VERSION%
    echo Please install a compatible Python version from https://python.org
    pause
    exit /b 1
)

if %PYTHON_MAJOR% EQU 3 (
    if %PYTHON_MINOR% LSS 10 (
        echo Error: Python 3.10-3.13 is required for Geti compatibility
        echo Found Python %PYTHON_VERSION%
        echo Please install a compatible Python version from https://python.org
        pause
        exit /b 1
    )
    if %PYTHON_MINOR% GTR 13 (
        echo Error: Python 3.10-3.13 is required for Geti compatibility
        echo Found Python %PYTHON_VERSION%
        echo Please install a compatible Python version from https://python.org
        pause
        exit /b 1
    )
)

if %PYTHON_MAJOR% GTR 3 (
    echo Error: Python 3.10-3.13 is required for Geti compatibility
    echo Found Python %PYTHON_VERSION%
    echo Please install a compatible Python version from https://python.org
    pause
    exit /b 1
)

echo [OK] Python %PYTHON_VERSION% is compatible with Geti requirements
echo.

echo Starting InferNode setup...
echo.

REM Create virtual environment if it doesn't exist
if not exist .venv (
    echo Creating virtual environment...
    python -m venv .venv
    if %errorlevel% neq 0 (
        echo Error: Failed to create virtual environment
        pause
        exit /b 1
    )
    echo Virtual environment created successfully.
    echo Creating .gitignore in .venv...
    echo * > .venv\.gitignore
) else (
    echo Virtual environment already exists.
)

echo.
echo Activating virtual environment...
call .venv\Scripts\activate.bat

echo.
echo Installing/upgrading pip...
python -m pip install --upgrade pip

echo.
echo Checking for CUDA and installing PyTorch with appropriate CUDA support...
REM Check for CUDA version using nvcc first, then nvidia-smi as fallback
nvcc --version >nul 2>&1
if %errorlevel% equ 0 (
    REM Extract CUDA version from nvcc output
    for /f "tokens=5" %%i in ('nvcc --version ^| findstr "release"') do set CUDA_VERSION=%%i
    REM Remove 'V' prefix if present and extract major.minor
    set CUDA_VERSION=%CUDA_VERSION:V=%
    for /f "tokens=1,2 delims=." %%a in ("%CUDA_VERSION%") do (
        set CUDA_MAJOR=%%a
        set CUDA_MINOR=%%b
    )
) else (
    REM Fallback to nvidia-smi if nvcc not found
    nvidia-smi --query-gpu=driver_version --format=csv,noheader,nounits >nul 2>&1
    if %errorlevel% equ 0 (
        REM If nvidia-smi works, assume CUDA is available but use CPU version as safe fallback
        echo CUDA drivers detected via nvidia-smi, but CUDA toolkit not found.
        echo Installing CPU-only PyTorch. For GPU support, install CUDA toolkit.
        set CUDA_SUFFIX=
    ) else (
        echo No CUDA installation detected. Installing CPU-only PyTorch.
        set CUDA_SUFFIX=
    )
)

REM Set PyTorch index URL based on CUDA version
if defined CUDA_MAJOR (
    if %CUDA_MAJOR% EQU 12 (
        if %CUDA_MINOR% EQU 6 (
            set PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu126
            echo Installing PyTorch with CUDA 12.6 support...
        ) else if %CUDA_MINOR% EQU 8 (
            set PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu128
            echo Installing PyTorch with CUDA 12.8 support...
        ) else if %CUDA_MINOR% EQU 9 (
            set PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu129
            echo Installing PyTorch with CUDA 12.9 support...
        ) else (
            set PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu121
            echo CUDA %CUDA_VERSION% detected. Using CUDA 12.1 PyTorch wheels as fallback...
        )
    ) else if %CUDA_MAJOR% EQU 11 (
        set PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu118
        echo Installing PyTorch with CUDA 11.8 support...
    ) else (
        set PYTORCH_INDEX_URL=
        echo Unsupported CUDA version %CUDA_VERSION%. Installing CPU-only PyTorch.
    )
) else (
    set PYTORCH_INDEX_URL=
    echo Installing CPU-only PyTorch.
)

REM Install PyTorch and torchvision with appropriate CUDA support
if defined PYTORCH_INDEX_URL (
    pip install torch torchvision --index-url %PYTORCH_INDEX_URL%
) else (
    pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
)

echo.
echo Installing dependencies from requirements.txt...
if exist "requirements.txt" (
    pip install -r requirements.txt
    if %errorlevel% neq 0 (
        echo Warning: Some dependencies may have failed to install
    )
) else (
    echo Warning: requirements.txt not found
)

echo.
echo Installing Intel Geti SDK (Python 3.10-3.13 only)...
if %PYTHON_MINOR% LEQ 13 (
    if exist "requirements-geti.txt" (
        pip install -r requirements-geti.txt
        if %errorlevel% neq 0 (
            echo Warning: Geti SDK installation failed, continuing...
        ) else (
            echo [OK] Geti SDK installed successfully
        )
    )
) else (
    echo [SKIP] Geti SDK not compatible with Python %PYTHON_VERSION% (requires 3.10-3.13)
)

echo.
echo Installing InferNode in development mode...
pip install -e .
if %errorlevel% neq 0 (
    echo Warning: Failed to install InferNode package
)

echo.
echo Verifying installation...
python -c "from InferenceNode._version import __version__; print('InferNode version: ' + __version__)" 2>nul
if %errorlevel% neq 0 (
    echo Warning: InferNode package verification failed
) else (
    echo [OK] InferNode package installed successfully
)

echo.
echo Setup completed!
echo.
echo IMPORTANT: To use InferNode, activate the virtual environment first:
echo    call .venv\Scripts\activate.bat
echo.
echo Then you can:
echo 1. Start InferNode: python main.py --production
echo 2. Start with custom port: python main.py --port 8080
echo 3. Disable discovery: python main.py --no-discovery
echo 4. Disable telemetry: python main.py --no-telemetry
echo.
echo Web interface will be available at: http://localhost:5555
echo.
echo NOTE: If Geti SDK was skipped due to Python version, you can:
echo - Install manually: pip install "geti-sdk>=2.11.0" (requires Python 3.10-3.13)
echo - Or use a compatible Python version and re-run setup
echo.
echo For Docker deployment, see DOCKER.md or run:
echo    docker-build.bat
echo.
echo To deactivate the virtual environment later, use: deactivate
echo.
pause