@echo off
REM TorchForge Windows Setup Script
REM Author: Anil Prasad

echo ========================================
echo TorchForge - Windows Setup
echo ========================================
echo.

echo Checking Python installation...
python --version
if %ERRORLEVEL% NEQ 0 (
    echo ERROR: Python not found. Please install Python 3.8+ from python.org
    pause
    exit /b 1
)

echo.
echo Creating virtual environment...
python -m venv venv
if %ERRORLEVEL% NEQ 0 (
    echo ERROR: Failed to create virtual environment
    pause
    exit /b 1
)

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

echo.
echo Upgrading pip...
python -m pip install --upgrade pip

echo.
echo Installing TorchForge...
pip install -e .

if %ERRORLEVEL% NEQ 0 (
    echo ERROR: Failed to install TorchForge
    pause
    exit /b 1
)

echo.
echo ========================================
echo Installation Complete!
echo ========================================
echo.
echo To get started:
echo   1. Activate virtual environment: venv\Scripts\activate.bat
echo   2. Run examples: python examples\comprehensive_examples.py
echo   3. Run tests: pytest tests\ -v
echo.
echo GitHub: https://github.com/anilprasad/torchforge
echo Documentation: See README.md and WINDOWS_GUIDE.md
echo.

pause