@echo off
setlocal
cd /d "%~dp0"
python -c "import yt_dlp, cv2, numpy" 2>nul
if errorlevel 1 (
  echo Installing the ADAM Video Dataset Collector requirements...
  python -m pip install -r "%~dp0requirements.txt"
  if errorlevel 1 goto :adam_dependency_error
)
python -c "import selenium, requests, PIL" 2>nul
if errorlevel 1 (
  echo Installing the Dataset Collector requirements for ADAM...
  python -m pip install -r "D:\Users\PlayRobloxAllDay\Desktop\Programs\GoogleImageDatasetCollector\requirements.txt"
  if errorlevel 1 goto :dependency_error
)
python -c "import datasets, diffusers, transformers, accelerate, torch, torchvision" 2>nul
if errorlevel 1 (
  echo Installing the DDPM training requirements for ADAM...
  python -m pip install -r "D:\Users\PlayRobloxAllDay\Desktop\Programs\DDPM\requirements.txt"
  if errorlevel 1 goto :ddpm_dependency_error
)
python main.py
if errorlevel 1 (
  echo.
  echo ADAM could not start. Install the requirements with:
  echo   python -m pip install -r requirements.txt
  echo.
  pause
)
endlocal
exit /b

:adam_dependency_error
echo.
echo ADAM could not install its Video Dataset Collector requirements.
echo Run this command with the same Python used to start ADAM:
echo   python -m pip install -r "%~dp0requirements.txt"
echo.
pause
endlocal
exit /b

:dependency_error
echo.
echo ADAM could not install the Dataset Collector requirements.
echo Run this command and then launch ADAM again:
echo   python -m pip install -r "D:\Users\PlayRobloxAllDay\Desktop\Programs\GoogleImageDatasetCollector\requirements.txt"
echo.
pause
endlocal
exit /b

:ddpm_dependency_error
echo.
echo ADAM could not install the DDPM training requirements.
echo Run this command and then launch ADAM again:
echo   python -m pip install -r "D:\Users\PlayRobloxAllDay\Desktop\Programs\DDPM\requirements.txt"
echo.
pause
endlocal
exit /b