setup_docuchat.bat

@echo off
setlocal enabledelayedexpansion

REM =============================================================================
REM DocuChat Agent - Windows Setup Script
REM =============================================================================
REM This script provides Windows setup for DocuChat Agent.
REM For full functionality, we recommend using WSL2 with the bash script.
REM
REM Usage:
REM   setup_docuchat.bat [OPTIONS]
REM
REM Options:
REM   /auto             Run automated setup
REM   /help             Show this help message
REM
REM Requirements:
REM   - Windows 10/11 with WSL2 enabled
REM   - Docker Desktop for Windows
REM   - Python 3.10+ installed
REM =============================================================================

set "SCRIPT_VERSION=1.0.0"
set "SCRIPT_NAME=DocuChat Windows Setup"

REM Colors (limited on Windows CMD)
set "RED=[91m"
set "GREEN=[92m"
set "YELLOW=[93m"
set "BLUE=[94m"
set "CYAN=[96m"
set "WHITE=[97m"
set "NC=[0m"

REM Unicode symbols (may not display correctly on all systems)
set "CHECK=✓"
set "CROSS=✗"
set "INFO=i"
set "WARN=!"

echo.
echo %WHITE%╔══════════════════════════════════════════════════════════════════════════════╗%NC%
echo %WHITE%║                                                                              ║%NC%
echo %WHITE%║  ██████╗  ██████╗  ██████╗██╗   ██╗ ██████╗██╗  ██╗ █████╗ ████████╗        ║%NC%
echo %WHITE%║  ██╔══██╗██╔═══██╗██╔════╝██║   ██║██╔════╝██║  ██║██╔══██╗╚══██╔══╝        ║%NC%
echo %WHITE%║  ██║  ██║██║   ██║██║     ██║   ██║██║     ███████║███████║   ██║           ║%NC%
echo %WHITE%║  ██║  ██║██║   ██║██║     ██║   ██║██║     ██╔══██║██╔══██║   ██║           ║%NC%
echo %WHITE%║  ██████╔╝╚██████╔╝╚██████╗╚██████╔╝╚██████╗██║  ██║██║  ██║   ██║           ║%NC%
echo %WHITE%║  ╚═════╝  ╚═════╝  ╚═════╝ ╚═════╝  ╚═════╝╚═╝  ╚═╝╚═╝  ╚═╝   ╚═╝           ║%NC%
echo %WHITE%║                    WINDOWS SETUP - Installation Helper                      ║%NC%
echo %WHITE%╚══════════════════════════════════════════════════════════════════════════════╝%NC%
echo.

REM Parse command line arguments
set "AUTO_MODE=false"
:parse_args
if "%~1"=="/auto" (
    set "AUTO_MODE=true"
    shift
    goto parse_args
)
if "%~1"=="/help" (
    goto show_help
)
if "%~1"=="/?" (
    goto show_help
)
if not "%~1"=="" (
    echo %RED%%CROSS% Unknown option: %1%NC%
    goto show_help
)

echo %CYAN%Welcome to DocuChat Agent Windows Setup!%NC%
echo.
echo %YELLOW%For the best experience, we recommend using WSL2 with our Linux setup script.%NC%
echo This Windows script provides basic setup and guidance.
echo.

REM Check for WSL2
echo %BLUE%%INFO% Checking for WSL2...%NC%
wsl --version >nul 2>&1
if %errorlevel% equ 0 (
    echo %GREEN%%CHECK% WSL2 is available%NC%
    echo.
    echo %CYAN%Would you like to run the full Linux setup script in WSL2? [Y/n]%NC%
    if "%AUTO_MODE%"=="false" (
        set /p "response="
        if /i "!response!"=="n" goto windows_setup
        if /i "!response!"=="no" goto windows_setup
    )

    echo %BLUE%%INFO% Running Linux setup script in WSL2...%NC%
    wsl bash -c "cd /mnt/c$(echo '%cd%' | tr '\' '/' | cut -c3-) && ./setup_docuchat.sh"
    if !errorlevel! equ 0 (
        echo.
        echo %GREEN%%CHECK% Setup completed successfully in WSL2!%NC%
        goto completion
    ) else (
        echo %RED%%CROSS% WSL2 setup failed. Falling back to Windows setup.%NC%
        goto windows_setup
    )
) else (
    echo %YELLOW%%WARN% WSL2 not found. Proceeding with Windows setup.%NC%
    goto windows_setup
)

:windows_setup
echo.
echo %BLUE%%INFO% Starting Windows setup...%NC%
echo.

REM Check Python
echo %BLUE%%INFO% Checking Python installation...%NC%
python --version >nul 2>&1
if %errorlevel% neq 0 (
    echo %RED%%CROSS% Python not found. Please install Python 3.10+ from https://python.org%NC%
    goto error_exit
)

for /f "tokens=2" %%i in ('python --version 2^>^&1') do set "PYTHON_VERSION=%%i"
echo %GREEN%%CHECK% Found Python %PYTHON_VERSION%%NC%

REM Check Python version (basic check)
python -c "import sys; exit(0 if sys.version_info >= (3, 10) else 1)" 2>nul
if %errorlevel% neq 0 (
    echo %RED%%CROSS% Python 3.10+ required. Current version: %PYTHON_VERSION%%NC%
    goto error_exit
)

REM Check Docker
echo %BLUE%%INFO% Checking Docker Desktop...%NC%
docker --version >nul 2>&1
if %errorlevel% neq 0 (
    echo %RED%%CROSS% Docker Desktop not found.%NC%
    echo Please install Docker Desktop from: https://www.docker.com/products/docker-desktop
    echo After installation, restart this script.
    goto error_exit
)
echo %GREEN%%CHECK% Docker Desktop found%NC%

REM Check if Docker is running
docker info >nul 2>&1
if %errorlevel% neq 0 (
    echo %YELLOW%%WARN% Docker Desktop is not running. Please start Docker Desktop and try again.%NC%
    goto error_exit
)
echo %GREEN%%CHECK% Docker Desktop is running%NC%

REM Create virtual environment
echo %BLUE%%INFO% Creating Python virtual environment...%NC%
if not exist "venv" (
    python -m venv venv
    if %errorlevel% neq 0 (
        echo %RED%%CROSS% Failed to create virtual environment%NC%
        goto error_exit
    )
    echo %GREEN%%CHECK% Virtual environment created%NC%
) else (
    echo %GREEN%%CHECK% Virtual environment already exists%NC%
)

REM Activate virtual environment and install dependencies
echo %BLUE%%INFO% Installing Python dependencies...%NC%
call venv\Scripts\activate.bat
python -m pip install --upgrade pip

if exist "requirements.txt" (
    pip install -r requirements.txt
    if %errorlevel% neq 0 (
        echo %RED%%CROSS% Failed to install requirements%NC%
        goto error_exit
    )
) else (
    pip install -e .
    if %errorlevel% neq 0 (
        echo %RED%%CROSS% Failed to install package%NC%
        goto error_exit
    )
)
echo %GREEN%%CHECK% Python dependencies installed%NC%

REM Install spaCy model
echo %BLUE%%INFO% Installing spaCy language model...%NC%
python -m spacy download en_core_web_sm
if %errorlevel% neq 0 (
    echo %YELLOW%%WARN% spaCy model installation failed (non-critical)%NC%
) else (
    echo %GREEN%%CHECK% spaCy model installed%NC%
)

REM Setup Neo4j container
echo %BLUE%%INFO% Setting up Neo4j database container...%NC%

REM Stop existing container if running
docker stop docuchat-neo4j >nul 2>&1
docker rm docuchat-neo4j >nul 2>&1

REM Create data directories
if not exist "data\neo4j\data" mkdir "data\neo4j\data"
if not exist "data\neo4j\logs" mkdir "data\neo4j\logs"
if not exist "data\neo4j\import" mkdir "data\neo4j\import"
if not exist "data\neo4j\plugins" mkdir "data\neo4j\plugins"

REM Start Neo4j container
echo %BLUE%%INFO% Starting Neo4j container (this may take a moment)...%NC%
docker run -d ^
    --name docuchat-neo4j ^
    --restart unless-stopped ^
    -p 7474:7474 ^
    -p 7687:7687 ^
    -v "%cd%\data\neo4j\data":/data ^
    -v "%cd%\data\neo4j\logs":/logs ^
    -v "%cd%\data\neo4j\import":/var/lib/neo4j/import ^
    -v "%cd%\data\neo4j\plugins":/plugins ^
    -e NEO4J_AUTH=neo4j/docuchat123 ^
    -e "NEO4J_PLUGINS=[\"apoc\"]" ^
    -e NEO4J_dbms_security_procedures_unrestricted=apoc.* ^
    -e NEO4J_dbms_memory_heap_initial__size=1G ^
    -e NEO4J_dbms_memory_heap_max__size=2G ^
    neo4j:5.15-community

if %errorlevel% neq 0 (
    echo %RED%%CROSS% Failed to start Neo4j container%NC%
    goto error_exit
)

echo %GREEN%%CHECK% Neo4j container started%NC%

REM Wait for Neo4j to be ready
echo %BLUE%%INFO% Waiting for Neo4j to start (this may take up to 60 seconds)...%NC%
set /a "attempts=0"
:neo4j_wait
set /a "attempts+=1"
timeout /t 2 /nobreak >nul
curl -s http://localhost:7474 >nul 2>&1
if %errorlevel% equ 0 (
    echo %GREEN%%CHECK% Neo4j is ready!%NC%
    goto neo4j_ready
)
if %attempts% lss 30 goto neo4j_wait

echo %RED%%CROSS% Neo4j failed to start within 60 seconds%NC%
docker logs docuchat-neo4j --tail 20
goto error_exit

:neo4j_ready

REM Create other data directories
echo %BLUE%%INFO% Creating data directories...%NC%
if not exist "data\documents" mkdir "data\documents"
if not exist "data\feedback" mkdir "data\feedback"
if not exist "data\graph_db" mkdir "data\graph_db"
if not exist "data\model_cache" mkdir "data\model_cache"
if not exist "data\vector_db" mkdir "data\vector_db"
if not exist "chroma" mkdir "chroma"

REM Test DocuChat CLI
echo %BLUE%%INFO% Testing DocuChat installation...%NC%
python -m docuchat --version >nul 2>&1
if %errorlevel% neq 0 (
    echo %RED%%CROSS% DocuChat CLI test failed%NC%
    goto error_exit
)

python -m docuchat --version
echo %GREEN%%CHECK% DocuChat CLI is working%NC%

REM Create test document
echo %BLUE%%INFO% Creating test documents...%NC%
if not exist "test_documents" mkdir "test_documents"
(
echo # DocuChat Test Document
echo.
echo This is a sample document for testing DocuChat Agent on Windows.
echo.
echo ## Key Features
echo - Document ingestion and processing
echo - Knowledge graph creation
echo - Natural language querying
echo - Local processing for privacy
echo.
echo ## Technology Stack
echo - Python 3.10+
echo - Neo4j for graph database
echo - ChromaDB for vector storage
echo - Local LLM integration
echo.
echo This document can be used to test the ingestion pipeline.
) > "test_documents\sample.txt"

echo %GREEN%%CHECK% Test documents created%NC%

goto completion

:show_help
echo.
echo %WHITE%%SCRIPT_NAME% v%SCRIPT_VERSION%%NC%
echo.
echo %CYAN%USAGE:%NC%
echo     %0 [OPTIONS]
echo.
echo %CYAN%OPTIONS:%NC%
echo     /auto             Run automated setup
echo     /help             Show this help message
echo.
echo %CYAN%REQUIREMENTS:%NC%
echo     - Windows 10/11
echo     - Python 3.10+ from https://python.org
echo     - Docker Desktop from https://docker.com
echo     - 8GB+ RAM recommended
echo     - 10GB+ free disk space
echo.
echo %CYAN%RECOMMENDED:%NC%
echo     For full functionality, enable WSL2 and use the Linux setup script:
echo     1. Install WSL2: wsl --install
echo     2. Restart your computer
echo     3. Run: ./setup_docuchat.sh
echo.
goto end

:completion
echo.
echo %GREEN%%CHECK% DocuChat Agent Windows setup completed!%NC%
echo.
echo %CYAN%Quick Start Guide:%NC%
echo.
echo %WHITE%1. Activate the Python environment:%NC%
echo    %CYAN%venv\Scripts\activate.bat%NC%
echo.
echo %WHITE%2. Test the installation:%NC%
echo    %CYAN%python -m docuchat --version%NC%
echo    %CYAN%python -m docuchat status%NC%
echo.
echo %WHITE%3. Ingest some documents:%NC%
echo    %CYAN%python -m docuchat ingest test_documents%NC%
echo.
echo %WHITE%4. Start chatting:%NC%
echo    %CYAN%python -m docuchat chat%NC%
echo.
echo %YELLOW%Services:%NC%
echo   • Neo4j Web UI: %CYAN%http://localhost:7474%NC% (neo4j/docuchat123)
echo.
echo %YELLOW%Important Notes:%NC%
echo   • For Ollama LLM support, install from: https://ollama.ai
echo   • For OCR support, install Tesseract: https://tesseract-ocr.github.io
echo   • Consider using WSL2 for full Linux compatibility
echo.
echo %WHITE%For help and documentation:%NC%
echo   %CYAN%https://github.com/your-org/docuchat-agent%NC%
echo.
goto end

:error_exit
echo.
echo %RED%%CROSS% Setup failed. Please check the error messages above.%NC%
echo.
echo %YELLOW%Common solutions:%NC%
echo   • Ensure Python 3.10+ is installed from python.org
echo   • Install Docker Desktop and ensure it's running
echo   • Check your internet connection
echo   • Run as Administrator if needed
echo.
echo %WHITE%For full support, consider using WSL2 with the Linux setup script.%NC%
echo.
exit /b 1

:end
pause
exit /b 0