我用bat做了一个AI任务自动化工具，但是一打开就闪退

xnssg

各位大佬好！我是刚接触批处理的小白自学，写了个AI工具脚本，但一运行就闪退，卡了实在没头绪😢 想请教大家帮忙看看怎么改～

1. @echo off
   
2. chcp 65001 >nul
   
3. setlocal enabledelayedexpansion
   
4. set "DEBUG_MODE=1"
   
5. 
   
6. :: AI任务自动化工具 v1.0.2
   
7. set "PROJECT_PATH=D:\AI_Project"
   
8. set "VENV_PATH=%PROJECT_PATH%\venv"
   
9. set "LOG_FILE=%PROJECT_PATH%\log\ai_tool.log"
   
10. set "CONFIG_FILE=%PROJECT_PATH%\config.ini"
    
11. set "INPUT_FILE=%PROJECT_PATH%\input.txt"
    
12. set "OUTPUT_FILE=%PROJECT_PATH%\output.txt"
    
13. 
    
14. set "LOG_LEVEL_DEBUG=1"
    
15. set "LOG_LEVEL_INFO=2"
    
16. set "LOG_LEVEL_WARN=3"
    
17. set "LOG_LEVEL_ERROR=4"
    
18. set "CURRENT_LOG_LEVEL=%LOG_LEVEL_INFO%"
    
19. 
    
20. if not exist "%PROJECT_PATH%\log" mkdir "%PROJECT_PATH%\log" 2>nul
    
21. 
    
22. :: 日志函数
    
23. :log
    
24. if %CURRENT_LOG_LEVEL% gtr %1 exit /b
    
25. for /f "skip=1" %%x in ('wmic os get LocalDateTime') do if not defined timestamp set "timestamp=%%x"
    
26. set "timestamp=%timestamp:~0,4%-%timestamp:~4,2%-%timestamp:~6,2% %timestamp:~8,2%:%timestamp:~10,2%:%timestamp:~12,2%"
    
27. if %1 equ %LOG_LEVEL_DEBUG% (set "level_prefix=DEBUG") else if %1 equ %LOG_LEVEL_INFO% (set "level_prefix=INFO ") else if %1 equ %LOG_LEVEL_WARN% (set "level_prefix=WARN ") else (set "level_prefix=ERROR")
    
28. echo %timestamp% [%level_prefix%] - %* >> "%LOG_FILE%"
    
29. if %DEBUG_MODE% equ 1 echo %timestamp% [%level_prefix%] - %*
    
30. exit /b
    
31. 
    
32. :: 检查Python
    
33. :check_python
    
34. echo 检查Python环境...
    
35. python --version >nul 2>&1 || (
    
36.     echo 错误: 未找到Python解释器！请安装Python并添加到PATH。
    
37.     echo 下载地址: https://www.python.org/downloads/
    
38.     call :log %LOG_LEVEL_ERROR% "未找到Python解释器"
    
39.     pause
    
40.     exit /b 1
    
41. )
    
42. exit /b 0
    
43. 
    
44. :: 创建目录
    
45. :create_dir
    
46. if not exist "%~1" (
    
47.     mkdir "%~1" 2>nul || (
    
48.         echo 错误: 无法创建目录 "%~1"！请检查权限。
    
49.         call :log %LOG_LEVEL_ERROR% "无法创建目录: %~1"
    
50.         pause
    
51.         exit /b 1
    
52.     )
    
53.     echo 已创建目录: %~1
    
54.     call :log %LOG_LEVEL_INFO% "已创建目录: %~1"
    
55. )
    
56. exit /b 0
    
57. 
    
58. :: 创建虚拟环境
    
59. :create_venv
    
60. echo 创建Python虚拟环境...
    
61. python -m venv "%VENV_PATH%" || (
    
62.     echo 错误: 无法创建虚拟环境！请检查Python安装。
    
63.     call :log %LOG_LEVEL_ERROR% "无法创建虚拟环境"
    
64.     pause
    
65.     exit /b 1
    
66. )
    
67. echo 虚拟环境已创建: %VENV_PATH%
    
68. call :log %LOG_LEVEL_INFO% "虚拟环境已创建"
    
69. exit /b 0
    
70. 
    
71. :: 安装依赖
    
72. :install_dependencies
    
73. echo 安装Python依赖库...
    
74. "%VENV_PATH%\Scripts\python.exe" -m pip install torch diffusers transformers -q || (
    
75.     echo 错误: 依赖库安装失败！请检查网络连接
    
76.     call :log %LOG_LEVEL_ERROR% "依赖库安装失败"
    
77.     pause
    
78.     exit /b 1
    
79. )
    
80. echo 依赖库安装完成
    
81. call :log %LOG_LEVEL_INFO% "依赖库安装完成"
    
82. exit /b 0
    
83. 
    
84. :: 初始化环境
    
85. :init_env
    
86. call :log %LOG_LEVEL_INFO% "开始环境初始化"
    
87. echo ==========================
    
88. echo        环境初始化
    
89. echo ==========================
    
90. echo.
    
91. 
    
92. call :check_python || exit /b 1
    
93. call :create_dir "%PROJECT_PATH%" || exit /b 1
    
94. call :create_dir "%PROJECT_PATH%\outputs"
    
95. call :create_dir "%PROJECT_PATH%\log"
    
96. 
    
97. if not exist "%VENV_PATH%\Scripts\python.exe" (
    
98.     call :create_venv || exit /b 1
    
99.     call :install_dependencies || exit /b 1
    
100. )
     
101. 
     
102. set "PYTHON_EXE=%VENV_PATH%\Scripts\python.exe"
     
103. echo 使用Python解释器: %PYTHON_EXE%
     
104. call :log %LOG_LEVEL_INFO% "使用Python解释器: %PYTHON_EXE%"
     
105. 
     
106. echo 检查CUDA环境...
     
107. %PYTHON_EXE% -c "import torch; print('CUDA可用' if torch.cuda.is_available() else 'CUDA不可用')" >nul 2>&1
     
108. if %errorlevel% neq 0 (
     
109.     echo 警告: 无法检测CUDA环境，图像生成可能无法使用GPU加速
     
110.     call :log %LOG_LEVEL_WARN% "无法检测CUDA环境"
     
111. ) else call :log %LOG_LEVEL_INFO% "CUDA环境检查完成"
     
112. 
     
113. call :log %LOG_LEVEL_INFO% "环境初始化完成"
     
114. echo 环境初始化完成！
     
115. timeout /t 2 >nul
     
116. goto menu
     
117. 
     
118. :: 主菜单
     
119. :menu
     
120. cls
     
121. echo =============================
     
122. echo    本地AI任务自动化工具 v1.0.2
     
123. echo =============================
     
124. echo.
     
125. echo  1. 图像生成（Stable Diffusion）
     
126. echo  2. 文本生成（LLaMA 7B）
     
127. echo  3. 查看日志
     
128. echo  4. 重新初始化环境
     
129. echo  5. 使用方法
     
130. echo  6. 退出程序
     
131. echo.
     
132. echo ========================
     
133. choice /c 123456 /n /m "请选择操作 [1-6]: "
     
134. if errorlevel 6 goto exit
     
135. if errorlevel 5 goto usage
     
136. if errorlevel 4 goto init_env
     
137. if errorlevel 3 goto view_log
     
138. if errorlevel 2 goto txt_gen_menu
     
139. if errorlevel 1 goto img_gen_menu
     
140. 
     
141. :: 图像生成
     
142. :img_gen_menu
     
143. cls
     
144. echo =====================================
     
145. echo             图像生成设置
     
146. echo =====================================
     
147. echo.
     
148. set /p prompt=请输入提示词:
     
149. set /p steps=生成步数 [50]:
     
150. if "%steps%"=="" set steps=50
     
151. set /p seed=随机种子 [1234]:
     
152. if "%seed%"=="" set seed=1234
     
153. 
     
154. echo.
     
155. echo 生成参数:
     
156. echo 提示词: %prompt%
     
157. echo 步数: %steps%
     
158. echo 种子: %seed%
     
159. echo.
     
160. choice /c yn /n /m "确认生成图像吗？(Y/N): "
     
161. if errorlevel 2 goto menu
     
162. 
     
163. for /f "skip=1" %%x in ('wmic os get LocalDateTime') do if not defined timestamp set "timestamp=%%x"
     
164. set "timestamp=%timestamp:~0,4%%timestamp:~4,2%%timestamp:~6,2%_%timestamp:~8,2%%timestamp:~10,2%%timestamp:~12,2%"
     
165. set "OUTPUT_DIR=%PROJECT_PATH%\outputs\%timestamp%"
     
166. call :create_dir "%OUTPUT_DIR%" || goto menu
     
167. 
     
168. call :log %LOG_LEVEL_INFO% "开始图像生成 - 提示词: %prompt%, 步数: %steps%, 种子: %seed%, 输出: %OUTPUT_DIR%"
     
169. 
     
170. echo 正在生成图像...
     
171. %PYTHON_EXE% generate_image.py --prompt "%prompt%" --steps %steps% --seed %seed% --output "%OUTPUT_DIR%" > "%OUTPUT_DIR%\generation.log" 2>&1
     
172. if %errorlevel% neq 0 (
     
173.     echo 错误: 生成失败！查看 %OUTPUT_DIR%\generation.log 获取详情
     
174.     call :log %LOG_LEVEL_ERROR% "图像生成失败 - 错误代码: %errorlevel%"
     
175.     if exist "%OUTPUT_DIR%\generation.log" type "%OUTPUT_DIR%\generation.log" | findstr /i "error" >> "%LOG_FILE%"
     
176.     pause
     
177. ) else (
     
178.     echo 图像生成成功！输出目录: %OUTPUT_DIR%
     
179.     call :log %LOG_LEVEL_INFO% "图像生成成功"
     
180.     timeout /t 3 >nul
     
181. )
     
182. goto menu
     
183. 
     
184. :: 文本生成
     
185. :txt_gen_menu
     
186. cls
     
187. echo ========================
     
188. echo           文本生成设置
     
189. echo ========================
     
190. echo.
     
191. echo 注意: 文本生成将从 %INPUT_FILE% 读取输入提示
     
192. echo.
     
193. set /p timeout=超时时间（秒） [30]:
     
194. if "%timeout%"=="" set timeout=30
     
195. echo.
     
196. echo 生成参数:
     
197. echo 输入文件: %INPUT_FILE%
     
198. echo 输出文件: %OUTPUT_FILE%
     
199. echo 超时: %timeout%秒
     
200. echo.
     
201. choice /c yn /n /m "确认开始文本生成吗？(Y/N): "
     
202. if errorlevel 2 goto menu
     
203. 
     
204. if not exist "%INPUT_FILE%" (
     
205.     echo 错误: 输入文件 "%INPUT_FILE%" 不存在！创建示例文件...
     
206.     >"%INPUT_FILE%" echo 请输入提示文本，AI将基于此生成内容。
     
207.     if exist "%INPUT_FILE%" (
     
208.         echo 已创建示例输入文件: %INPUT_FILE%
     
209.         call :log %LOG_LEVEL_WARN% "创建示例输入文件"
     
210.         timeout /t 3 >nul
     
211.     ) else (
     
212.         echo 错误: 无法创建示例输入文件！
     
213.         call :log %LOG_LEVEL_ERROR% "无法创建示例输入文件"
     
214.         pause
     
215.         goto menu
     
216.     )
     
217. )
     
218. 
     
219. (
     
220.     echo [text_gen]
     
221.     echo input_file=%INPUT_FILE%
     
222.     echo output_file=%OUTPUT_FILE%
     
223.     echo timeout=%timeout%
     
224. ) > "%CONFIG_FILE%" 2>nul || (
     
225.     echo 错误: 无法创建配置文件！
     
226.     call :log %LOG_LEVEL_ERROR% "无法创建配置文件: %CONFIG_FILE%"
     
227.     pause
     
228.     goto menu
     
229. )
     
230. 
     
231. call :log %LOG_LEVEL_INFO% "开始文本生成 - 输入: %INPUT_FILE%, 输出: %OUTPUT_FILE%, 超时: %timeout%秒"
     
232. 
     
233. echo 正在生成文本...
     
234. %PYTHON_EXE% generate_text.py --config "%CONFIG_FILE%" > "%PROJECT_PATH%\outputs\text_generation.log" 2>&1
     
235. if %errorlevel% neq 0 (
     
236.     echo 错误: 生成失败！查看 %PROJECT_PATH%\outputs\text_generation.log 获取详情
     
237.     call :log %LOG_LEVEL_ERROR% "文本生成失败 - 错误代码: %errorlevel%"
     
238.     if exist "%PROJECT_PATH%\outputs\text_generation.log" type "%PROJECT_PATH%\outputs\text_generation.log" | findstr /i "error" >> "%LOG_FILE%"
     
239.     pause
     
240. ) else (
     
241.     echo 文本生成成功！输出文件: %OUTPUT_FILE%
     
242.     call :log %LOG_LEVEL_INFO% "文本生成成功"
     
243.     timeout /t 3 >nul
     
244. )
     
245. goto menu
     
246. 
     
247. :: 查看日志
     
248. :view_log
     
249. cls
     
250. echo ===========================
     
251. echo           操作日志
     
252. echo ===========================
     
253. echo.
     
254. 
     
255. if not exist "%LOG_FILE%" (
     
256.     echo 日志文件不存在或为空。
     
257.     pause
     
258.     goto menu
     
259. )
     
260. 
     
261. set "page_size=15"
     
262. for /f %%a in ('type "%LOG_FILE%" ^| find /c /v ""') do set /a "total_lines=%%a, pages=(total_lines+page_size-1)/page_size"
     
263. if %pages% equ 0 set pages=1
     
264. 
     
265. set /p page=输入页数 [1-%pages%]:
     
266. if "%page%"=="" set page=1
     
267. if %page% lss 1 (echo 错误: 页数不能小于1！& timeout /t 2 >nul & goto view_log)
     
268. if %page% gtr %pages% (echo 错误: 页数不能大于%pages%！& timeout /t 2 >nul & goto view_log)
     
269. 
     
270. set /a start_line=(%page%-1)*%page_size%+1, end_line=start_line+page_size-1
     
271. if %end_line% gtr %total_lines% set end_line=%total_lines%
     
272. 
     
273. echo 第%page%页 / 共%pages%页
     
274. echo.
     
275. 
     
276. setlocal enabledelayedexpansion
     
277. set "lines=0"
     
278. for /f "usebackq delims=" %%a in ("%LOG_FILE%") do (
     
279.     set /a lines+=1
     
280.     if !lines! geq %start_line% if !lines! leq %end_line% echo !line!
     
281. )
     
282. endlocal
     
283. 
     
284. echo.
     
285. echo =========================================
     
286. pause >nul
     
287. goto menu
     
288. 
     
289. :: 使用方法
     
290. :usage
     
291. cls
     
292. echo ==========================
     
293. echo           使用方法
     
294. echo ==========================
     
295. echo.
     
296. echo 1. 环境初始化：自动检查并初始化环境
     
297. echo 2. 图像生成：基于Stable Diffusion生成图像
     
298. echo 3. 文本生成：基于LLaMA 7B生成文本
     
299. echo 4. 重新初始化环境：重建虚拟环境并安装依赖
     
300. echo 5. 查看日志：查看工具运行记录
     
301. echo 6. 退出程序：结束工具运行
     
302. echo.
     
303. pause >nul
     
304. goto menu
     
305. 
     
306. :: 退出程序
     
307. :exit
     
308. call :log %LOG_LEVEL_INFO% "程序正常退出"
     
309. cls
     
310. echo =================================
     
311. echo    感谢使用本地AI任务自动化工具！
     
312. echo =================================
     
313. timeout /t 2 >nul
     
314. exit /b 0
     
315. 
     
316. :: 直接运行时初始化环境
     
317. if "%0"=="%~dpnx0" call :init_env

复制代码

flashercs

bat的标签是按顺序执行的,:log :check_python :create_dir
应该把:menu 放到前面

Batcher

回复 2# flashercs


请参考Q-01观察一下在哪一步退出的：
https://mp.weixin.qq.com/s/Koje4FufWxWBs7ioDy_LJA

aloha20200628

回复 1# xnssg

一楼代码至少有两点须特别注意》
一。代码先切换码页到65001，就要求存盘批处脚本文件时须采用utf-8编码，否则代码中的中文屏显数据会变成乱码...
二。批处脚本是从上往下逐行（或逐块）执行的，因此，主控部分的行序要先于那些被其调用的子过程或子函数（除非用条件goto跳过），否则后者会在未被前者调用前就会执行，当然容易出错。例如代码中的 :log 子过程应该后置到代码末尾。据此，楼主须重新理清每个 :xxx 子过程与主控部分的从属（调用）关系及其行序...

Five66

运行完第30行就直接退出了吧
那么多exit /b而且又没有用goto跳过 ，大概是流程没控制好
还有代码里用了chcp 65001 >nul ，bat文件编码得保存为utf-8，最后是没有bom的，而且为了防止utf-8编码出问题，最好在包含中文的那些代码行末尾添加一个英文空格

xnssg

回复 5# Five66

谢谢啦

xnssg

回复 4# aloha20200628


    谢谢啦

xnssg

回复 3# Batcher
谢谢啦

xnssg

回复 2# flashercs


    谢谢啦

jiavip

该说的楼上的大神们都说得差不多了，
告诉你一个经验之谈，
有个万能的查错方法：在感觉有可能出错的地方插入pause.
特别是在:xxx 标签结尾句或跳跃句，插入，，

学会灵活的使用pause暂停查错方法，基本可以自行排查脚本编写过程中一大半以上的出错。特别适用小白和新手，，
也是一步步拆解脚本运行步骤的重要方法，没有任何复杂的脚本能抵挡得了，一步步暂停之后被拆解的过程。