调整信息抽取方法的生成位置

config/keys.py

@@ -22,3 +22,12 @@ HOSPITAL = ["医院"]
 DEPARTMENT = ["科室"]
 # 主治医生
 DOCTOR = ["主治医生"]
+
+SETTLEMENT_LIST_SCHEMA = \
+    PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES + PERSONAL_CASH_PAYMENT \
+    + PERSONAL_ACCOUNT_PAYMENT + PERSONAL_FUNDED_AMOUNT + MEDICAL_INSURANCE_TYPE
+
+DISCHARGE_RECORD_SCHEMA = \
+    HOSPITAL + DEPARTMENT + PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + DOCTOR
+
+COST_LIST_SCHEMA = PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES

config/photo_review.py

@@ -1,3 +1,7 @@
+from paddlenlp import Taskflow
+
+from config.keys import SETTLEMENT_LIST_SCHEMA, DISCHARGE_RECORD_SCHEMA, COST_LIST_SCHEMA
+
 # 项目奔溃的重试次数
 RETRY_TIME = 1
 

photo_review/__init__.py

@@ -1,25 +0,0 @@
-from paddlenlp import Taskflow
-
-from config.keys import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \
-    PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR
-from config.photo_review import IE_BATCH_SIZE, LAYOUT_ANALYSIS
-
-# 基本医保结算单
-settlement_list_schema = \
-    PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES + PERSONAL_CASH_PAYMENT \
-    + PERSONAL_ACCOUNT_PAYMENT + PERSONAL_FUNDED_AMOUNT + MEDICAL_INSURANCE_TYPE
-settlement_ie = Taskflow("information_extraction", schema=settlement_list_schema, model="uie-x-base",
-                         task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS,
-                         batch_size=IE_BATCH_SIZE)
-
-# 出院记录
-discharge_record_schema = \
-    HOSPITAL + DEPARTMENT + PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + DOCTOR
-discharge_ie = Taskflow("information_extraction", schema=discharge_record_schema, model="uie-x-base",
-                        task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS,
-                        batch_size=IE_BATCH_SIZE)
-
-# 费用清单
-cost_list_schema = PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES
-cost_ie = Taskflow("information_extraction", schema=cost_list_schema, model="uie-x-base",
-                   task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS, batch_size=IE_BATCH_SIZE)

photo_review/photo_review.py

@@ -19,8 +19,7 @@ from sqlalchemy import update
 from config.keys import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \
     PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR
 from config.mysql import MysqlSession
-from config.photo_review import PHHD_BATCH_SIZE, SLEEP_MINUTES
+from config.photo_review import PHHD_BATCH_SIZE, SLEEP_MINUTES, SETTLEMENT_IE, DISCHARGE_IE, COST_IE
-from photo_review import settlement_ie, discharge_ie, cost_ie
 from photo_review.entity.bd_yljg import BdYljg
 from photo_review.entity.bd_ylks import BdYlks
 from photo_review.entity.zx_ie_cost import ZxIeCost
@@ -220,7 +219,7 @@ def photo_review(pk_phhd):
         elif phrec.cRectype == "4":
             cost_list.append(phrec)
 
-    settlement_list_ie_result = information_extraction(settlement_ie, settlement_list)
+    settlement_list_ie_result = information_extraction(SETTLEMENT_IE, settlement_list)
     settlement_data = {
         "pk_phhd": pk_phhd,
         "name": handle_name(get_best_value_in_keys(settlement_list_ie_result, PATIENT_NAME)),
@@ -246,7 +245,7 @@ def photo_review(pk_phhd):
     settlement_data["personal_funded_amount"] = handle_decimal(settlement_data["personal_funded_amount_str"])
     save_or_update_ie(ZxIeSettlement, pk_phhd, settlement_data)
 
-    discharge_record_ie_result = information_extraction(discharge_ie, discharge_record)
+    discharge_record_ie_result = information_extraction(DISCHARGE_IE, discharge_record)
     discharge_data = {
         "pk_phhd": pk_phhd,
         "hospital": get_best_value_in_keys(discharge_record_ie_result, HOSPITAL),
@@ -283,7 +282,7 @@ def photo_review(pk_phhd):
                 discharge_data["department"] = ylks.name
     save_or_update_ie(ZxIeDischarge, pk_phhd, discharge_data)
 
-    cost_list_ie_result = information_extraction(cost_ie, cost_list)
+    cost_list_ie_result = information_extraction(COST_IE, cost_list)
     cost_data = {
         "pk_phhd": pk_phhd,
         "name": handle_name(get_best_value_in_keys(cost_list_ie_result, PATIENT_NAME)),

visual_model_test/__init__.py

@@ -1 +1 @@
-import photo_review
+