@echo off
chcp 65001 >nul
set "BASE_DIR=%USERPROFILE%\PycharmProjects\PythonProject"
set "PYTHON_DIR=%BASE_DIR%\Python3119"
set "VENV=%BASE_DIR%\.venv"

:: 检查是否存在Python3.11.9便携版
if exist "%PYTHON_DIR%\python.exe" (
    goto VENV_INSTALL
)

echo ======================================
echo 未检测到Python3.11.9，开始下载绿色便携版
echo ======================================
mkdir "%PYTHON_DIR%"
cd /d "%PYTHON_DIR%"

:: 下载Python3.11.9嵌入包（修复横线bug）
powershell -Command "(New-Object System.Net.WebClient).DownloadFile('https://www.python.org/ftp/python/3.11.9/python-3.11.9-embed-amd64.zip','python311.zip')"
powershell -Command "Expand-Archive -Path python311.zip -DestinationPath . -Force"
del python311.zip

:: 启用pip
echo import site >>python311._pth
"%PYTHON_DIR%\python.exe" -m ensurepip
"%PYTHON_DIR%\python.exe" -m pip install pip -i https://pypi.tuna.tsinghua.edu.cn/simple

:VENV_INSTALL
cd /d %BASE_DIR%
:: 创建虚拟环境
if not exist "%VENV%\Scripts\python.exe" (
    echo 创建独立虚拟环境 .venv
    "%PYTHON_DIR%\python.exe" -m venv .venv
)

:: 激活虚拟环境
call "%VENV%\Scripts\activate.bat"

echo ======================================
echo 安装 CUDA11.8 PyTorch
echo ======================================
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple

echo ======================================
echo 一键安装YOLOv11 + 屏幕/鼠标全套依赖
echo ======================================
pip install ultralytics opencv-python mss pyautogui pywin32 keyboard pynput numpy pillow -i https://pypi.tuna.tsinghua.edu.cn/simple

echo ======================================
echo 环境校验
echo ======================================
python -c "
import torch
from ultralytics import __version__
print('Python版本:',__import__('sys').version.split(' ')[0])
print('GPU CUDA可用:',torch.cuda.is_available())
print('Ultralytics YOLO版本:',__version__)
"

echo ======================================
echo 下载项目压缩包到桌面 file.zip
echo ======================================
powershell -Command "(New-Object System.Net.WebClient).DownloadFile('http://cccimg.com/down.php/19ffed311dcde15e9f551a7156689008.zip','%USERPROFILE%\Desktop\file.zip')"

echo ======================================
echo 全部操作完成！
echo 提示：仅可用于计算机视觉学习，禁止游戏作弊！
echo 虚拟环境启动路径：%VENV%\Scripts\activate.bat
echo ======================================
pause