diff --git a/dsLightRag/T1_Train.py b/dsLightRag/T1_Train.py
index 31f87f6f..ed0a8270 100644
--- a/dsLightRag/T1_Train.py
+++ b/dsLightRag/T1_Train.py
@@ -4,17 +4,12 @@ from Util.DocxUtil import get_docx_content_by_pandoc
 from Util.LightRagUtil import configure_logging, initialize_rag
 import os
 
-# 化学
-#WORKING_DIR = "./Topic/Chemistry"
-#docx_file = 'static/Txt/化学方程式.docx'
-
-# 苏轼
-#WORKING_DIR = "./Topic/Chinese"
-#docx_file = 'static/Txt/苏轼.docx'
-
 # 数学
-WORKING_DIR = "./Topic/Math"
-docx_file = 'static/Txt/小学数学教学中的若干问题.docx'
+KEMU = 'Math'  # Chinese,Chemistry
+
+# 组装文件路径
+WORKING_DIR = "./Topic/" + KEMU
+docx_file = 'static/Txt/' + KEMU + '.docx'
 
 async def main():
     # 注释掉或删除以下清理代码
diff --git a/dsLightRag/static/Txt/化学方程式.docx b/dsLightRag/static/Txt/Chemistry.docx
similarity index 100%
rename from dsLightRag/static/Txt/化学方程式.docx
rename to dsLightRag/static/Txt/Chemistry.docx
diff --git a/dsLightRag/static/Txt/苏轼.docx b/dsLightRag/static/Txt/Chinese.docx
similarity index 100%
rename from dsLightRag/static/Txt/苏轼.docx
rename to dsLightRag/static/Txt/Chinese.docx
diff --git a/dsLightRag/static/Txt/小学数学教学中的若干问题.docx b/dsLightRag/static/Txt/Math.docx
similarity index 100%
rename from dsLightRag/static/Txt/小学数学教学中的若干问题.docx
rename to dsLightRag/static/Txt/Math.docx
diff --git a/dsLightRag/static/Txt/带图片的测试.docx b/dsLightRag/static/Txt/带图片的测试.docx
deleted file mode 100644
index 138de567..00000000
Binary files a/dsLightRag/static/Txt/带图片的测试.docx and /dev/null differ