fix:添加UI界面,完善功能

This commit is contained in:
2025-12-11 11:16:09 +08:00
parent 81e3c40abb
commit f437842a81
17 changed files with 939 additions and 406 deletions

View File

@@ -10,12 +10,13 @@ from config.config import load_config
config = load_config("config.toml")
def generate_comment(name, age_group, traits):
def generate_comment(name, age_group, traits,sex):
"""
生成评语
:param name: 学生姓名
:param age_group: 所在班级
:param traits: 表现特征
:param sex: 性别
:return: 评语
"""
@@ -29,7 +30,7 @@ def generate_comment(name, age_group, traits):
# 2. 构建 Prompt Template
prompt = ChatPromptTemplate.from_messages([
("system", ai_config["prompt"]),
("human", "学生姓名:{name}\n所在班级:{age_group}\n表现特征:{traits}\n\n请开始撰写评语:")
("human", "学生姓名:{name}\n所在班级:{age_group}\n性别:{sex}\n表现特征:{traits}\n\n请开始撰写评语:")
])
# 3. 组装链 (Prompt -> Model -> OutputParser)
@@ -40,11 +41,12 @@ def generate_comment(name, age_group, traits):
comment = chain.invoke({
"name": name,
"age_group": age_group,
"traits": traits
"traits": traits,
"sex": sex
})
cleaned_text = re.sub(r'\s+', '', comment)
logger.success(f"学生:{name} =>生成评语成功: {cleaned_text}")
return cleaned_text
except Exception as e:
print(f"生成评语失败: {e}")
logger.error(f"生成评语失败: {e}")
return "生成失败请检查网络或Key。"