根据疾病名生成病例prompt

prompt

请根据疾病名：" + disease_name + " 为我生成一份病历。下面是病历内容的要求：病例应严格包含如下几项: 性别，年龄，疾病名（必须是" + disease_name + "），主诉（不超过20字），现病史（包括症状起因、治疗经过等），既往史（包括慢性疾病史、传染病史等），个人史（包括生活习惯、工作环境等），家族史，体格检查（包括体温、脉搏、呼吸、血压等），辅助检查（疾病相关的检测项目，但是检查项目必须和疾病相关），初步诊断（需要明确结果，内容需要包含疾病名）等11项， 要求如下：不要出现假设，字数要大于50字。 一定要包含性别，年龄，疾病名，主诉，现病史，既往史，个人史，家族史，体格检查，辅助检查，初步诊断11个部分且字段名严格保持一致。参考格式如下：【性别】：男\n【年龄】：35岁\n【疾病名】：" + disease_name + "\n【主诉】\n不超过20字的症状疾病描述，症状需要有伴随时间，比如右脚大拇指疼痛，活动受限1个月。\n【现病史】\n现患疾病有哪些\n【既往史】\n过去生过哪些病。\n【个人史】\n吸烟饮酒工作生活睡眠等状况\n【家族史】\n家族中无类似疾病患者\n【体格检查】\n和" + disease_name + "病相关的身体外形检查\n【辅助检查】\n和" + disease_name + "病相关的体检指标\n【初步诊断】\n急性支气管炎\n"

代码：

import os
import pandas as pdimport warnings
warnings.filterwarnings("ignore")
from tqdm import tqdm
from peft import AutoPeftModelForCausalLM
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfigqwen_path = "/mnt/sdd/Qwen-7B-Chat"
qwen_tokenizer = AutoTokenizer.from_pretrained(qwen_path, trust_remote_code=True)
qwen_model = AutoModelForCausalLM.from_pretrained(qwen_path, device_map="auto", trust_remote_code=True).eval()def read_file():root_path = "/home/wangyp/Big_Model/symptom_3_explain/disease_name"all_data = []# 遍历文件夹下的文件for filename in os.listdir(root_path):with open(os.path.join(root_path, filename), mode="r", encoding="utf-8") as f:print(os.path.join(root_path, filename))hh = (f.readlines())all_data = all_data + hha1 = list(set(all_data))a2 = [a.replace("'", "").strip() for a in a1]print(len(a2))print(a2[:5])return a2def qwen(disease_name):"""年龄，性别，疾病名（必须和我提供的疾病名一致），主诉，现病史，既往史，个人史，家族史，体格检查，辅助检查（需要明确结果，最好有描述，最起码有结果），初步诊断（需要明确结果，最好有描述，最起码有结果）等11项，:param disease_name::return:"""# pmt = "你现在扮演一名医生的角色，我需要你基于我给的疾病名："+ disease_name +"，生成对应的【病历】，病例应严格包含如下几项，不能出现其他幻想项目:患者年龄、性别、疾病名称（必须和我提供的疾病名一致）、主诉、患者病史、诊断方案（需要明确结果，最好有描述，最起码有结果）、查体、辅助检查、诊断依据、鉴别诊断等10个部分，要求如下：诊断要和病例非常匹配，不要出现假设，描述详细且专业、字数要大于50字、且一定要有疾病名称这个部分"# pmt = "你现在扮演一名医生的角色，我需要你基于我给的疾病名："+ disease_name + "，生成对应的【病历】，病例应严格包含如下几项: 性别，年龄，疾病名（必须和我提供的疾病名一致），主诉（包括症状描述和持续时间），现病史（包括症状起因、治疗经过等），既往史（包括慢性疾病史、传染病史等），个人史（包括生活习惯、工作环境等），家族史，体格检查（包括体温、脉搏、呼吸、血压等），辅助检查（疾病相关的检测项目，但是检查项目必须和疾病相关），初步诊断（需要明确结果，内容需要包含疾病名）等11项， 要求如下：不要出现假设，描述详细且专业、字数要大于50字。 一定要包含性别，年龄，疾病名，主诉，现病史，既往史，个人史，家族史，体格检查，辅助检查，初步诊断11个部分且字段名严格保持一致。参考格式如下：【性别】：男\n【年龄】：35岁\n【疾病名】：急性支气管炎\n【主诉】\n患者自诉近一个月来反复出现咳嗽、咳痰，痰为白色黏液状，伴有胸闷、气短。近三天症状加重，出现低热，体温波动在37.5-38.0℃之间。无咯血、胸痛、呼吸困难等其他症状。\n【现病史】\n患者平素体健，无慢性疾病史。近一个月来因工作压力增大，作息不规律，开始出现咳嗽、咳痰症状。曾自行服用止咳药（具体不详），症状未见缓解。近三天症状加重，并出现低热。患者自发病以来，精神尚可，食欲欠佳，睡眠不佳，大小便正常，体重无明显变化。\n【既往史】\n否认高血压、糖尿病、冠心病等慢性疾病史。否认肝炎、结核等传染病史。否认手术、输血史。否认药物过敏史。预防接种史不详。\n【个人史】\n患者出生于本地，无外地旅居史。居住环境良好，无吸烟、饮酒等不良嗜好。工作压力大，作息不规律，缺乏运动。\n【家族史】\n家族中无类似疾病患者，无遗传性疾病史。\n【体格检查】\n体温：37.8℃脉搏：80次/分呼吸：20次/分血压：120/80mmHg\n一般情况可，神清语明，面色红润。肺部听诊可闻及双肺散在湿啰音。心腹查体未见明显异常。\n【辅助检查】\n血常规：白细胞计数轻度升高，中性粒细胞比例增高。\n胸部X线片：双肺纹理增粗，可见散在斑片状阴影。\n【初步诊断】\n急性支气管炎\n"pmt = "请根据疾病名：" + disease_name + "为我生成一份病历。下面是病历内容的要求：病例应严格包含如下几项: 性别，年龄，疾病名（必须是" + disease_name + "），主诉（不超过20字），现病史（包括症状起因、治疗经过等），既往史（包括慢性疾病史、传染病史等），个人史（包括生活习惯、工作环境等），家族史，体格检查（包括体温、脉搏、呼吸、血压等），辅助检查（疾病相关的检测项目，但是检查项目必须和疾病相关），初步诊断（需要明确结果，内容需要包含疾病名）等11项， 要求如下：不要出现假设，字数要大于50字。 一定要包含性别，年龄，疾病名，主诉，现病史，既往史，个人史，家族史，体格检查，辅助检查，初步诊断11个部分且字段名严格保持一致。参考格式如下：【性别】：男\n【年龄】：35岁\n【疾病名】：" + disease_name + "\n【主诉】\n不超过20字的症状疾病描述，症状需要有伴随时间，比如右脚大拇指疼痛，活动受限1个月。\n【现病史】\n现患疾病有哪些\n【既往史】\n过去生过哪些病。\n【个人史】\n吸烟饮酒工作生活睡眠等状况\n【家族史】\n家族中无类似疾病患者\n【体格检查】\n和" + disease_name + "病相关的身体外形检查\n【辅助检查】\n和" + disease_name + "病相关的体检指标\n【初步诊断】\n急性支气管炎\n"response, history = qwen_model.chat(qwen_tokenizer, pmt, history=None)return (response)"""
传入dict，
遍历dict, 如果这个dict没有生成出来，就继续生成，直到不以抱歉为开头，结束这个循环继续下一个
传出dict
"""
def find_exp_item(dict):for disease_name, value in tqdm(dict.items()):if "抱歉" in value:print(disease_name)pmt = "请根据疾病名：" + disease_name + "为我生成一份病历。下面是病历内容的要求：病例应严格包含如下几项: 性别，年龄，疾病名（必须是" + disease_name + "），主诉（不超过20字），现病史（包括症状起因、治疗经过等），既往史（包括慢性疾病史、传染病史等），个人史（包括生活习惯、工作环境等），家族史，体格检查（包括体温、脉搏、呼吸、血压等），辅助检查（疾病相关的检测项目，但是检查项目必须和疾病相关），初步诊断（需要明确结果，内容需要包含疾病名）等11项， 要求如下：不要出现假设，字数要大于50字。 一定要包含性别，年龄，疾病名，主诉，现病史，既往史，个人史，家族史，体格检查，辅助检查，初步诊断11个部分且字段名严格保持一致。参考格式如下：【性别】：男\n【年龄】：35岁\n【疾病名】：" + disease_name + "\n【主诉】\n不超过20字的症状疾病描述，症状需要有伴随时间，比如右脚大拇指疼痛，活动受限1个月。\n【现病史】\n现患疾病有哪些\n【既往史】\n过去生过哪些病。\n【个人史】\n吸烟饮酒工作生活睡眠等状况\n【家族史】\n家族中无类似疾病患者\n【体格检查】\n和" + disease_name + "病相关的身体外形检查\n【辅助检查】\n和" + disease_name + "病相关的体检指标\n【初步诊断】\n急性支气管炎\n"response, history = qwen_model.chat(qwen_tokenizer, pmt, history=None)i = 0while ("抱歉" in response and i < 3):print(".................再次尝试" + disease_name)response, history = qwen_model.chat(qwen_tokenizer, pmt, history=None)i += 1#  此刻退出，复制继续下一轮循环dict[disease_name] = responseelse:continuereturn dictif __name__ == '__main__':# print("********************运行开始**********************")# li = read_file()# li = ["外阴皮赘"]## f2 = []## for i in tqdm(li):#     ll = qwen(i)#     print(ll)#     f2.append(ll)# df = pd.DataFrame({"疾病名": li, "病历": f2 })# df.to_csv('/home/wangyp/Big_Model/symptom_3_explain/disease_name/data.csv', index=False)# print("********************运行结束**********************")print("********************运行开始**********************")df = pd.read_csv("/home/wangyp/Big_Model/symptom_3_explain/disease_name/data.csv", index_col=0).to_dict()dict = df["病历"]# dict = {#     '肛周感染': '抱歉，由于肛周感染是一种常见的细菌性炎症性疾病，通常不会引发明显的全身症状，因此无法生成具有详细病历信息的示例。建议您向医生咨询以获取准确的病历信息。',#     '急性淋巴细胞性白血病（L1型）': '【性别】：女\n【年龄】：48岁\n【疾病名】：急性淋巴细胞性白血病（L1型）\n【主诉】：近一周来，乏力、消瘦、发热，伴淋巴结肿大。\n【现病史】：1年前曾有过感冒，目前未接种疫苗。\n【既往史】：高血压病史，已服用降压药。\n【个人史】：经常熬夜，饮食不规律，工作压力大。\n【家族史】：无类似疾病患者。\n【体格检查】：体温37.5℃，心率80次/分，呼吸20次/分，血压140/90mmHg。颈部可见多个淋巴结肿大。\n【辅助检查】：血液检查显示白细胞计数增高，骨髓穿刺示淋巴细胞增殖明显。\n【初步诊断】：急性淋巴细胞性白血病（L1型）。建议进一步完善检查并行化疗。',#     '传导性失语': '抱歉，【性别】：男\n【年龄】：35岁\n【疾病名】：传导性失语\n【主诉】 已经持续一周的左耳听力下降，伴有言语模糊不清。\n【现病史】 一周前开始感觉左耳听力下降，随着时间推移，语言表达逐渐困难，最近几天已经无法正常沟通。\n【既往史】 患有高血压、糖尿病，最近没有感冒发热等症状。\n【个人史】 吸烟3年，饮酒适度，工作环境良好，作息规律，每天保证7小时以上的睡眠。\n【家族史】 父母无此类疾病，兄弟姐妹均身体健康。\n【体格检查】 体温正常，脉搏稳定在78次/分钟，呼吸正常，血压140/90mmHg。\n【辅助检查】 耳鼻喉科检查显示左侧鼓膜充血，内耳淋巴液肿胀明显；血液检查显示血糖偏高，血压波动较大。\n【初步诊断】 由于长期吸烟饮酒，导致左耳听神经受损，引起传导性失语。建议戒烟限酒，加强营养，药物治疗，并进行康复训练。'}res = find_exp_item(dict)df = pd.DataFrame({"疾病名": res.keys(), "病历": res.values() })df.to_csv('/home/wangyp/Big_Model/symptom_3_explain/disease_name/data_complete.csv', index=False)print("********************运行结束**********************")

结果