@echo off
REM Fix PyTorch Installation for Lemontropia Suite
REM Run this as Administrator

echo ==========================================
echo Fixing PyTorch for Computer Vision
echo ==========================================
echo.

REM Step 1: Uninstall broken packages
echo [1/4] Uninstalling broken PyTorch packages...
pip uninstall -y torch torchvision torchaudio paddlepaddle-gpu paddlepaddle paddleocr 2>nul

REM Step 2: Clean pip cache
echo [2/4] Cleaning pip cache...
pip cache purge 2>nul

REM Step 3: Install CPU-only PyTorch (most stable on Windows Store Python)
echo [3/4] Installing CPU-only PyTorch (stable version)...
pip install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cpu

REM Step 4: Install PaddleOCR (CPU version)
echo [4/4] Installing PaddleOCR...
pip install paddlepaddle==2.5.2
pip install paddleocr==2.7.0.3

REM Step 5: Verify installation
echo.
echo ==========================================
echo Verifying installation...
echo ==========================================
python -c "import torch; print(f'PyTorch: {torch.__version__}')"
python -c "import torch; print(f'CUDA available: {torch.cuda.is_available()}')"
python -c "from paddleocr import PaddleOCR; print('PaddleOCR: OK')"

echo.
echo ==========================================
echo Installation Complete!
echo ==========================================
echo.
echo If you see errors above, try running:
echo   pip install --upgrade pip setuptools wheel
@echo Then run this script again.
@pause