fix:添加UI界面,完善功能
This commit is contained in:
@@ -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。"
|
||||
|
||||
Reference in New Issue
Block a user