@echo off
echo Creating virtual environment...
python -m venv .xraymodels

echo.
echo Activating environment and installing packages...
call .xraymodels\Scripts\activate && (
    echo Upgrade pip...
    python -m pip install --no-cache-dir --upgrade pip
    echo Environment activated successfully
    python --version
    echo.
    echo Installing remaining requirements...
    uv pip install --no-cache-dir --upgrade numpy==1.26.4 pandas Pillow requests tqdm wget evaluate rouge nltk scikit_learn ftfy spacy huggingface_hub tensorboardX diffusers protobuf "transformers>=4.35.0" "gradio>=4.0.0"
    echo.
    echo Installing PyTorch with CUDA support...
    pip install --no-cache-dir --upgrade torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
    echo Installation complete!
) || (
    echo Failed to activate environment
    pause
)