param(
    [string]$Python = "py -3.10"
)

$ErrorActionPreference = "Stop"
$root = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
$venv = Join-Path $root ".venv-katib-ocr"

if (-not (Test-Path $venv)) {
    Invoke-Expression "$Python -m venv `"$venv`""
}

$pythonExe = Join-Path $venv "Scripts\python.exe"
& $pythonExe -m pip install --upgrade pip
& $pythonExe -m pip install -r (Join-Path $root "requirements-katib-ocr.txt")
& $pythonExe -c "import torch, torchvision, transformers; from transformers import AutoModelForImageTextToText, AutoProcessor"

Write-Host "KATIB Arabic OCR sidecar is ready at $venv"
Write-Host "Default model: oddadmix/Katib-Qwen3.5-0.8B-0.1"
Write-Host "Use OCR_ENGINE=katib-ocr for smaller Arabic-trained OCR on short samples or a worker."