@echo off
CHCP 65001 > nul
setlocal enabledelayedexpansion

echo 你需要选择启动方式，输入字母来选择:
echo   V = 不知道什么意思就输入 V
echo   C = 输入 C 使用 Conda 环境
echo.
choice /C CV /N /M "在下方输入字母并回车 (C/V)?" /T 10 /D V

set "ENV_TYPE="
if %ERRORLEVEL% == 1 set "ENV_TYPE=CONDA"
if %ERRORLEVEL% == 2 set "ENV_TYPE=VENV"

if "%ENV_TYPE%" == "CONDA" (
    set /p CONDA_ENV_NAME="请输入要使用的 Conda 环境名称: "
    if not defined CONDA_ENV_NAME (
        echo 错误: 未输入 Conda 环境名称.
        pause
        exit /b 1
    )
    echo 选择: Conda '%CONDA_ENV_NAME%'
    REM 激活Conda环境
    call conda activate %CONDA_ENV_NAME%
    if %ERRORLEVEL% neq 0 (
        echo 错误: Conda环境 '%CONDA_ENV_NAME%' 激活失败. 请确保Conda已安装并正确配置, 且 '%CONDA_ENV_NAME%' 环境存在.
        pause
        exit /b 1
    )
) else (
    echo Selected: venv (default)
    REM 查找venv虚拟环境
    set "venv_path=%~dp0venv\Scripts\activate.bat"
    if not exist "%venv_path%" (
        echo Error: venv not found. Ensure the venv directory exists alongside the script.
        pause
        exit /b 1
    )
    REM 激活虚拟环境
    call "%venv_path%"
    if %ERRORLEVEL% neq 0 (
        echo Error: Failed to activate venv virtual environment.
        pause
        exit /b 1
    )
)

echo Environment activated successfully!

REM --- 后续脚本执行 ---

REM 运行预处理脚本
python "%~dp0scripts\raw_data_preprocessor.py"
if %ERRORLEVEL% neq 0 (
    echo Error: raw_data_preprocessor.py execution failed.
    pause
    exit /b 1
)

REM 运行信息提取脚本
python "%~dp0scripts\info_extraction.py"
if %ERRORLEVEL% neq 0 (
    echo Error: info_extraction.py execution failed.
    pause
    exit /b 1
)

REM 运行OpenIE导入脚本
python "%~dp0scripts\import_openie.py"
if %ERRORLEVEL% neq 0 (
    echo Error: import_openie.py execution failed.
    pause
    exit /b 1
)

echo All processing steps completed!
pause