@echo off
title Scan2Doc Pro v6.0 - Installer
color 0A
echo.
echo  ==========================================
echo   Scan2Doc Pro v6.0 - Installer
echo   Created by Us.Abolfazl
echo  ==========================================
echo.

set INSTALL_DIR=%USERPROFILE%\Scan2Doc_Pro

echo [1/5] Creating folder: %INSTALL_DIR%
if not exist "%INSTALL_DIR%" mkdir "%INSTALL_DIR%"

echo [2/5] Copying files...
copy /y "%~dp0scan2doc_pro_v6.py" "%INSTALL_DIR%\" >nul
copy /y "%~dp0scan2doc_gui_pro.py" "%INSTALL_DIR%\" >nul
if exist "%~dp0icon.ico" copy /y "%~dp0icon.ico" "%INSTALL_DIR%\" >nul
if exist "%~dp0icon.png" copy /y "%~dp0icon.png" "%INSTALL_DIR%\" >nul
echo  Files copied!

echo [3/5] Installing Python packages...
pip install opencv-python numpy pytesseract python-docx Pillow pymupdf >nul 2>&1
if errorlevel 1 (
    echo  [WARNING] Some packages failed. Trying python -m pip...
    python -m pip install opencv-python numpy pytesseract python-docx Pillow pymupdf >nul 2>&1
)
echo  Packages installed!

echo [4/5] Checking Tesseract...
where tesseract >nul 2>&1
if errorlevel 1 (
    echo  Tesseract not in PATH. Trying winget...
    winget install UB-Mannheim.TesseractOCR --silent --accept-package-agreements --accept-source-agreements >nul 2>&1
    if errorlevel 1 (
        echo.
        echo  [!] Tesseract not installed automatically.
        echo  Please install manually:
        echo  https://github.com/UB-Mannheim/tesseract/wiki
        echo  Or run: winget install UB-Mannheim.TesseractOCR
        echo.
    ) else (
        echo  Tesseract installed via winget!
    )
) else (
    echo  Tesseract found!
)

echo [5/5] Creating launchers...

echo @echo off > "%INSTALL_DIR%\Scan2Doc_GUI.bat"
echo cd /d "%INSTALL_DIR%" >> "%INSTALL_DIR%\Scan2Doc_GUI.bat"
echo python scan2doc_gui_pro.py >> "%INSTALL_DIR%\Scan2Doc_GUI.bat"
echo pause >> "%INSTALL_DIR%\Scan2Doc_GUI.bat"

echo @echo off > "%INSTALL_DIR%\Scan2Doc_CLI.bat"
echo cd /d "%INSTALL_DIR%" >> "%INSTALL_DIR%\Scan2Doc_CLI.bat"
echo python scan2doc_pro_v6.py %%* >> "%INSTALL_DIR%\Scan2Doc_CLI.bat"
echo pause >> "%INSTALL_DIR%\Scan2Doc_CLI.bat"

echo [DONE] Creating desktop shortcut...
set DESKTOP=%USERPROFILE%\Desktop
set VBS=%TEMP%\create_shortcut.vbs
(
echo Set WshShell = CreateObject^("WScript.Shell"^)
echo Set shortcut = WshShell.CreateShortcut^("%DESKTOP%\Scan2Doc Pro.lnk"^)
echo shortcut.TargetPath = "%INSTALL_DIR%\Scan2Doc_GUI.bat"
echo shortcut.WorkingDirectory = "%INSTALL_DIR%"
echo shortcut.Description = "Scan2Doc Pro v6.0"
echo shortcut.Save
) > "%VBS%"
cscript //nologo "%VBS%" 2>nul
del "%VBS%" 2>nul

echo.
echo  ==========================================
echo   INSTALLATION COMPLETE!
echo  ==========================================
echo.
echo  Folder:    %INSTALL_DIR%
echo  Shortcut:  Scan2Doc Pro (Desktop)
echo.
echo  How to use:
echo    GUI:  Double-click "Scan2Doc Pro" on Desktop
echo    CLI:  python scan2doc_pro_v6.py -i image.jpg -o output.docx
echo.
echo  Created by Us.Abolfazl
echo.
pause