@echo off
REM Batch script to check if all dependencies are installed
REM Run this script from the project root directory

echo =========================================
echo Dependency Checker for SAM2 Image Auto Segment
echo =========================================
echo.

REM Check if Python is installed
python --version >nul 2>&1
if errorlevel 1 (
    echo [ERROR] Python is not installed or not in PATH!
    echo Please install Python 3.10 or higher
    pause
    exit /b 1
)
python --version
echo.

echo Checking installed packages...
echo.

REM List of required packages from requirements.txt
set MISSING_PACKAGES=
set FAILED_IMPORTS=

REM Check each package
echo [1/14] Checking fastapi...
python -c "import fastapi; print('  OK - fastapi', fastapi.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% fastapi

echo [2/14] Checking uvicorn...
python -c "import uvicorn; print('  OK - uvicorn', uvicorn.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% uvicorn

echo [3/14] Checking numpy...
python -c "import numpy; print('  OK - numpy', numpy.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% numpy

echo [4/14] Checking opencv-python...
python -c "import cv2; print('  OK - opencv-python', cv2.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% opencv-python

echo [5/14] Checking Pillow...
python -c "import PIL; print('  OK - Pillow', PIL.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% Pillow

echo [6/14] Checking scikit-image...
python -c "import skimage; print('  OK - scikit-image', skimage.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% scikit-image

echo [7/14] Checking torch...
python -c "import torch; print('  OK - torch', torch.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% torch

echo [8/14] Checking torchvision...
python -c "import torchvision; print('  OK - torchvision', torchvision.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% torchvision

echo [9/14] Checking huggingface_hub...
python -c "import huggingface_hub; print('  OK - huggingface_hub', huggingface_hub.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% huggingface_hub

echo [10/14] Checking omegaconf...
python -c "import omegaconf; print('  OK - omegaconf', omegaconf.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% omegaconf

echo [11/14] Checking hydra-core...
python -c "import hydra; print('  OK - hydra-core', hydra.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% hydra-core

echo [12/14] Checking iopath...
python -c "import iopath; print('  OK - iopath')" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% iopath

echo [13/14] Checking tqdm...
python -c "import tqdm; print('  OK - tqdm', tqdm.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% tqdm

echo [14/14] Checking requests...
python -c "import requests; print('  OK - requests', requests.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% requests

echo [15/14] Checking psutil...
python -c "import psutil; print('  OK - psutil', psutil.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% psutil

echo.
echo Checking SAM2 package...
python -c "import sam2; print('  OK - sam2 package is installed')" 2>nul || (
    echo [WARNING] SAM2 package is not installed
    echo   Run: cd sam2 ^&^& python -m pip install -e . ^&^& cd ..
    set MISSING_PACKAGES=%MISSING_PACKAGES% sam2
)

echo.
echo =========================================
if "%MISSING_PACKAGES%"=="" (
    echo All dependencies are installed! ✓
    echo.
    echo Testing critical imports...
    python -c "from model.sam_model import predict_polygon; print('  OK - model.sam_model')" 2>nul || echo [WARNING] Cannot import model.sam_model
    python -c "from model.utils import load_image_from_url; print('  OK - model.utils')" 2>nul || echo [WARNING] Cannot import model.utils
    python -c "from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator; print('  OK - sam2.automatic_mask_generator')" 2>nul || echo [WARNING] Cannot import sam2.automatic_mask_generator
    echo.
    echo ✓ Ready to run the application!
) else (
    echo Missing packages:%MISSING_PACKAGES%
    echo.
    echo Please install missing packages by running:
    echo   install_requirements.bat
    echo.
    echo Or manually:
    echo   python -m pip install -r requirements.txt
    echo   cd sam2
    echo   python -m pip install -e .
    echo   cd ..
)
echo =========================================
echo.
pause