优化三项资料分类方法

photo_review/auto_photo_review.py

@@ -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': '基本医保结算单',