优化三项资料分类方法

This commit is contained in:
2024-10-18 13:04:42 +08:00
parent 6529dc3d98
commit 401954dca0

View File

@@ -146,7 +146,14 @@ def information_extraction(phrec, pk_phhd):
ocr_result += tmp_ocr_result
tmp_ocr_text = common_util.ocr_result_to_text(ocr_result)
tmp_rec_type = model_util.clas_text(tmp_ocr_text) if ocr_text else None
if any(key in tmp_ocr_text for key in ['出院记录', '出院小结', '死亡记录']):
tmp_rec_type = '出院记录'
elif any(key in tmp_ocr_text for key in ['费用汇总清单', '费用清单', '费用明细', '结账清单', '费用小项统计']):
tmp_rec_type = '费用清单'
elif any(key in tmp_ocr_text for key in ['住院收费票据', '结算单', '财政部监制', '结算凭证']):
tmp_rec_type = '基本医保结算单'
else:
tmp_rec_type = model_util.clas_text(tmp_ocr_text) if tmp_ocr_text else None
if not tmp_rec_type:
rec_dict = {
'1': '基本医保结算单',