修正docker镜像构建
This commit is contained in:
@@ -24,7 +24,7 @@ services:
|
|||||||
context: ./services/paddle_services
|
context: ./services/paddle_services
|
||||||
container_name: ocr_api
|
container_name: ocr_api
|
||||||
hostname: ocr_api
|
hostname: ocr_api
|
||||||
command: [ '-w 1 ./services/ocr_api:app --bind 0.0.0.0:5001' ]
|
command: [ '-w 1 ocr_api:app --bind 0.0.0.0:5001' ]
|
||||||
deploy:
|
deploy:
|
||||||
resources:
|
resources:
|
||||||
reservations:
|
reservations:
|
||||||
|
|||||||
@@ -20,9 +20,9 @@ from db.mysql import BdYljg, BdYlks, ZxIeResult, ZxIeCost, ZxIeDischarge, ZxIeSe
|
|||||||
from log import HOSTNAME
|
from log import HOSTNAME
|
||||||
from photo_review import PHHD_BATCH_SIZE, SLEEP_MINUTES, HOSPITAL_ALIAS, HOSPITAL_FILTER, DEPARTMENT_ALIAS, \
|
from photo_review import PHHD_BATCH_SIZE, SLEEP_MINUTES, HOSPITAL_ALIAS, HOSPITAL_FILTER, DEPARTMENT_ALIAS, \
|
||||||
DEPARTMENT_FILTER
|
DEPARTMENT_FILTER
|
||||||
from services import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \
|
from services.paddle_services import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, \
|
||||||
PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR, \
|
PERSONAL_CASH_PAYMENT, PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, \
|
||||||
ADMISSION_ID, SETTLEMENT_ID, AGE, UPPERCASE_MEDICAL_EXPENSES
|
DEPARTMENT, DOCTOR, ADMISSION_ID, SETTLEMENT_ID, AGE, UPPERCASE_MEDICAL_EXPENSES
|
||||||
from ucloud import ufile
|
from ucloud import ufile
|
||||||
from util import image_util, common_util, html_util, model_util
|
from util import image_util, common_util, html_util, model_util
|
||||||
from util.data_util import handle_date, handle_decimal, parse_department, handle_name, handle_insurance_type, \
|
from util.data_util import handle_date, handle_decimal, parse_department, handle_name, handle_insurance_type, \
|
||||||
|
|||||||
@@ -1,34 +0,0 @@
|
|||||||
"""
|
|
||||||
信息抽取关键词配置
|
|
||||||
"""
|
|
||||||
|
|
||||||
# 患者姓名
|
|
||||||
PATIENT_NAME = ['患者姓名']
|
|
||||||
# 入院日期
|
|
||||||
ADMISSION_DATE = ['入院日期']
|
|
||||||
# 出院日期
|
|
||||||
DISCHARGE_DATE = ['出院日期']
|
|
||||||
# 发生医疗费
|
|
||||||
MEDICAL_EXPENSES = ['费用总额']
|
|
||||||
# 个人现金支付
|
|
||||||
PERSONAL_CASH_PAYMENT = ['个人现金支付']
|
|
||||||
# 个人账户支付
|
|
||||||
PERSONAL_ACCOUNT_PAYMENT = ['个人账户支付']
|
|
||||||
# 个人自费金额
|
|
||||||
PERSONAL_FUNDED_AMOUNT = ['自费金额', '个人自费']
|
|
||||||
# 医保类别
|
|
||||||
MEDICAL_INSURANCE_TYPE = ['医保类型']
|
|
||||||
# 就诊医院
|
|
||||||
HOSPITAL = ['医院']
|
|
||||||
# 就诊科室
|
|
||||||
DEPARTMENT = ['科室']
|
|
||||||
# 主治医生
|
|
||||||
DOCTOR = ['主治医生']
|
|
||||||
# 住院号
|
|
||||||
ADMISSION_ID = ['住院号']
|
|
||||||
# 医保结算单号码
|
|
||||||
SETTLEMENT_ID = ['医保结算单号码']
|
|
||||||
# 年龄
|
|
||||||
AGE = ['年龄']
|
|
||||||
# 大写总额
|
|
||||||
UPPERCASE_MEDICAL_EXPENSES = ['大写总额']
|
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ ENV PYTHONUNBUFFERED=1 \
|
|||||||
PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple
|
PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple
|
||||||
|
|
||||||
# 安装依赖
|
# 安装依赖
|
||||||
COPY ./services/paddle_services/requestments.txt /app/requirements.txt
|
COPY requestments.txt /app/requirements.txt
|
||||||
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
|
||||||
&& pip install --no-cache-dir -r requirements.txt \
|
&& pip install --no-cache-dir -r requirements.txt \
|
||||||
&& pip uninstall -y onnxruntime onnxruntime-gpu \
|
&& pip uninstall -y onnxruntime onnxruntime-gpu \
|
||||||
|
|||||||
@@ -0,0 +1,34 @@
|
|||||||
|
"""
|
||||||
|
信息抽取关键词配置
|
||||||
|
"""
|
||||||
|
|
||||||
|
# 患者姓名
|
||||||
|
PATIENT_NAME = ['患者姓名']
|
||||||
|
# 入院日期
|
||||||
|
ADMISSION_DATE = ['入院日期']
|
||||||
|
# 出院日期
|
||||||
|
DISCHARGE_DATE = ['出院日期']
|
||||||
|
# 发生医疗费
|
||||||
|
MEDICAL_EXPENSES = ['费用总额']
|
||||||
|
# 个人现金支付
|
||||||
|
PERSONAL_CASH_PAYMENT = ['个人现金支付']
|
||||||
|
# 个人账户支付
|
||||||
|
PERSONAL_ACCOUNT_PAYMENT = ['个人账户支付']
|
||||||
|
# 个人自费金额
|
||||||
|
PERSONAL_FUNDED_AMOUNT = ['自费金额', '个人自费']
|
||||||
|
# 医保类别
|
||||||
|
MEDICAL_INSURANCE_TYPE = ['医保类型']
|
||||||
|
# 就诊医院
|
||||||
|
HOSPITAL = ['医院']
|
||||||
|
# 就诊科室
|
||||||
|
DEPARTMENT = ['科室']
|
||||||
|
# 主治医生
|
||||||
|
DOCTOR = ['主治医生']
|
||||||
|
# 住院号
|
||||||
|
ADMISSION_ID = ['住院号']
|
||||||
|
# 医保结算单号码
|
||||||
|
SETTLEMENT_ID = ['医保结算单号码']
|
||||||
|
# 年龄
|
||||||
|
AGE = ['年龄']
|
||||||
|
# 大写总额
|
||||||
|
UPPERCASE_MEDICAL_EXPENSES = ['大写总额']
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from . import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES
|
|||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
COST_LIST_SCHEMA = PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES
|
COST_LIST_SCHEMA = PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES
|
||||||
COST = Taskflow('information_extraction', schema=COST_LIST_SCHEMA, model='uie-x-base',
|
COST = Taskflow('information_extraction', schema=COST_LIST_SCHEMA, model='uie-x-base',
|
||||||
task_path='../model/cost_list_model', layout_analysis=False, precision='fp16')
|
task_path='../../model/cost_list_model', layout_analysis=False, precision='fp16')
|
||||||
|
|
||||||
|
|
||||||
@app.route('/nlp/cost', methods=['POST'])
|
@app.route('/nlp/cost', methods=['POST'])
|
||||||
@@ -3,7 +3,7 @@ import os.path
|
|||||||
import cv2
|
import cv2
|
||||||
from flask import Flask, request
|
from flask import Flask, request
|
||||||
|
|
||||||
from paddle_services.paddle_detection import detector
|
from paddle_detection import detector
|
||||||
from utils import process_request, parse_img_path
|
from utils import process_request, parse_img_path
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
@@ -3,7 +3,7 @@ import os
|
|||||||
import cv2
|
import cv2
|
||||||
from flask import Flask, request
|
from flask import Flask, request
|
||||||
|
|
||||||
from paddle_services.doc_dewarp import dewarper
|
from doc_dewarp import dewarper
|
||||||
from utils import process_request, parse_img_path
|
from utils import process_request, parse_img_path
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
@@ -11,7 +11,7 @@ DISCHARGE_RECORD_SCHEMA = (
|
|||||||
HOSPITAL + DEPARTMENT + PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + DOCTOR + ADMISSION_ID + AGE
|
HOSPITAL + DEPARTMENT + PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + DOCTOR + ADMISSION_ID + AGE
|
||||||
)
|
)
|
||||||
DISCHARGE = Taskflow('information_extraction', schema=DISCHARGE_RECORD_SCHEMA, model='uie-x-base',
|
DISCHARGE = Taskflow('information_extraction', schema=DISCHARGE_RECORD_SCHEMA, model='uie-x-base',
|
||||||
task_path='../model/discharge_record_model', layout_analysis=False, precision='fp16')
|
task_path='../../model/discharge_record_model', layout_analysis=False, precision='fp16')
|
||||||
|
|
||||||
|
|
||||||
@app.route('/nlp/discharge', methods=['POST'])
|
@app.route('/nlp/discharge', methods=['POST'])
|
||||||
@@ -15,7 +15,7 @@ SETTLEMENT_LIST_SCHEMA = (
|
|||||||
+ UPPERCASE_MEDICAL_EXPENSES
|
+ UPPERCASE_MEDICAL_EXPENSES
|
||||||
)
|
)
|
||||||
SETTLEMENT_IE = Taskflow('information_extraction', schema=SETTLEMENT_LIST_SCHEMA, model='uie-x-base',
|
SETTLEMENT_IE = Taskflow('information_extraction', schema=SETTLEMENT_LIST_SCHEMA, model='uie-x-base',
|
||||||
task_path='../model/settlement_list_model', layout_analysis=False, precision='fp16')
|
task_path='../../model/settlement_list_model', layout_analysis=False, precision='fp16')
|
||||||
|
|
||||||
|
|
||||||
@app.route('/nlp/settlement', methods=['POST'])
|
@app.route('/nlp/settlement', methods=['POST'])
|
||||||
@@ -1,4 +1,3 @@
|
|||||||
import logging
|
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from flask import jsonify
|
from flask import jsonify
|
||||||
@@ -14,7 +13,6 @@ def process_request(func):
|
|||||||
result = func(*args, **kwargs)
|
result = func(*args, **kwargs)
|
||||||
return jsonify(result), 200
|
return jsonify(result), 200
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.getLogger('error').error(f'Error: {e}', exc_info=e)
|
|
||||||
return jsonify({'error': str(e)}), 500
|
return jsonify({'error': str(e)}), 500
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
Reference in New Issue
Block a user