From a859f9238fa0df18023317276dcce3c1f0d1da23 Mon Sep 17 00:00:00 2001
From: SengokuCola <1026294844@qq.com>
Date: Fri, 2 May 2025 19:17:59 +0800
Subject: [PATCH] =?UTF-8?q?better=EF=BC=9A=E4=BC=98=E5=8C=96=E5=8F=AF?=
 =?UTF-8?q?=E8=AF=BB=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ![新版麦麦开始学习.bat           | 65 +++++++++++++++++++++++---------
 scripts/info_extraction.py       |  4 +-
 scripts/raw_data_preprocessor.py |  2 +-
 3 files changed, 51 insertions(+), 20 deletions(-)

diff --git a/![新版麦麦开始学习.bat b/![新版麦麦开始学习.bat
index 41fc98368..b95bad004 100644
--- a/![新版麦麦开始学习.bat
+++ b/![新版麦麦开始学习.bat
@@ -2,26 +2,57 @@
 CHCP 65001 > nul
 setlocal enabledelayedexpansion
 
-REM 查找venv虚拟环境
-set "venv_path=%~dp0venv\Scripts\activate.bat"
-if not exist "%venv_path%" (
-    echo 错误: 未找到虚拟环境，请确保venv目录存在
-    pause
-    exit /b 1
+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
+    )
 )
 
-REM 激活虚拟环境
-call "%venv_path%"
-if %ERRORLEVEL% neq 0 (
-    echo 错误: 虚拟环境激活失败
-    pause
-    exit /b 1
-)
+echo Environment activated successfully!
+
+REM --- 后续脚本执行 ---
 
 REM 运行预处理脚本
 python "%~dp0scripts\raw_data_preprocessor.py"
 if %ERRORLEVEL% neq 0 (
-    echo 错误: raw_data_preprocessor.py 执行失败
+    echo Error: raw_data_preprocessor.py execution failed.
     pause
     exit /b 1
 )
@@ -29,7 +60,7 @@ if %ERRORLEVEL% neq 0 (
 REM 运行信息提取脚本
 python "%~dp0scripts\info_extraction.py"
 if %ERRORLEVEL% neq 0 (
-    echo 错误: info_extraction.py 执行失败
+    echo Error: info_extraction.py execution failed.
     pause
     exit /b 1
 )
@@ -37,10 +68,10 @@ if %ERRORLEVEL% neq 0 (
 REM 运行OpenIE导入脚本
 python "%~dp0scripts\import_openie.py"
 if %ERRORLEVEL% neq 0 (
-    echo 错误: import_openie.py 执行失败
+    echo Error: import_openie.py execution failed.
     pause
     exit /b 1
 )
 
-echo 所有处理步骤完成!
+echo All processing steps completed!
 pause
\ No newline at end of file
diff --git a/scripts/info_extraction.py b/scripts/info_extraction.py
index 65c4082b6..9e079070b 100644
--- a/scripts/info_extraction.py
+++ b/scripts/info_extraction.py
@@ -87,8 +87,8 @@ def main():
     signal.signal(signal.SIGINT, signal_handler)
 
     # 新增用户确认提示
-    print("=== 重要操作确认 ===")
-    print("实体提取操作将会花费较多资金和时间，建议在空闲时段执行。")
+    print("=== 重要操作确认，请认真阅读以下内容哦 ===")
+    print("实体提取操作将会花费较多api余额和时间，建议在空闲时段执行。")
     print("举例：600万字全剧情，提取选用deepseek v3 0324，消耗约40元，约3小时。")
     print("建议使用硅基流动的非Pro模型")
     print("或者使用可以用赠金抵扣的Pro模型")
diff --git a/scripts/raw_data_preprocessor.py b/scripts/raw_data_preprocessor.py
index 2fc30352e..33d51153f 100644
--- a/scripts/raw_data_preprocessor.py
+++ b/scripts/raw_data_preprocessor.py
@@ -44,7 +44,7 @@ def process_text_file(file_path):
 
 def main():
     # 新增用户确认提示
-    print("=== 重要操作确认 ===")
+    print("=== 重要操作确认，请认真阅读以下内容哦 ===")
     print("如果你并非第一次导入知识")
     print("请先删除data/import.json文件，备份data/openie.json文件")
     print("在进行知识库导入之前")