@echo off
echo ============================================================
echo Neural Network Trading System - Environment Setup
echo ============================================================

call conda activate gpt-gpu

echo Checking and installing required packages...

REM Check for PyTorch
python -c "import torch" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing PyTorch...
    call conda install -y pytorch torchvision cpuonly -c pytorch
)

REM Check for NumPy
python -c "import numpy" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing NumPy...
    call conda install -y numpy
)

REM Check for Pandas
python -c "import pandas" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing Pandas...
    call conda install -y pandas
)

REM Check for Matplotlib
python -c "import matplotlib" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing Matplotlib...
    call conda install -y matplotlib
)

REM Check for scikit-learn
python -c "import sklearn" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing scikit-learn...
    call conda install -y scikit-learn
)

REM Check for additional dependencies
python -c "import h5py" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing h5py...
    call conda install -y h5py
)

python -c "import tqdm" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing tqdm...
    call conda install -y tqdm
)

python -c "import yaml" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing PyYAML...
    call conda install -y pyyaml
)

python -c "import plotly" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing Plotly...
    call conda install -y plotly
)

python -c "import tensorboard" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing TensorBoard...
    call conda install -y tensorboard
)

python -c "import ccxt" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing ccxt...
    call pip install ccxt
)

echo ============================================================
echo Environment setup completed.
echo You can now run the Neural Network with: run_pytorch_nn.bat
echo ============================================================