From 7a4cb5263a176221e322f9456945ea12ab5b4629 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Fri, 27 Sep 2024 15:31:11 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=96=87=E6=9C=AC=E4=BF=A1?=
 =?UTF-8?q?=E6=81=AF=E6=8A=BD=E5=8F=96=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 services/paddle_services/ie_cost.py       |  7 ++++
 services/paddle_services/ie_discharge.py  |  7 ++++
 services/paddle_services/ie_settlement.py |  7 ++++
 util/model_util.py                        | 48 +++++++++++++++++++++++
 4 files changed, 69 insertions(+)

diff --git a/services/paddle_services/ie_cost.py b/services/paddle_services/ie_cost.py
index 446fbc9..5032e2e 100644
--- a/services/paddle_services/ie_cost.py
+++ b/services/paddle_services/ie_cost.py
@@ -22,6 +22,13 @@ def main():
     return COST({'doc': img_path, 'layout': json.loads(layout)})
 
 
+@app.route('/text', methods=['POST'])
+@process_request
+def text():
+    t = request.form.get('text')
+    return COST(t)
+
+
 if __name__ == '__main__':
     logging.config.dictConfig(LOGGING_CONFIG)
     app.run('0.0.0.0', 5004)
diff --git a/services/paddle_services/ie_discharge.py b/services/paddle_services/ie_discharge.py
index a7960e6..28702ba 100644
--- a/services/paddle_services/ie_discharge.py
+++ b/services/paddle_services/ie_discharge.py
@@ -24,6 +24,13 @@ def main():
     return DISCHARGE({'doc': img_path, 'layout': json.loads(layout)})
 
 
+@app.route('/text', methods=['POST'])
+@process_request
+def text():
+    t = request.form.get('text')
+    return DISCHARGE(t)
+
+
 if __name__ == '__main__':
     logging.config.dictConfig(LOGGING_CONFIG)
     app.run('0.0.0.0', 5003)
diff --git a/services/paddle_services/ie_settlement.py b/services/paddle_services/ie_settlement.py
index aa9d667..a160ff9 100644
--- a/services/paddle_services/ie_settlement.py
+++ b/services/paddle_services/ie_settlement.py
@@ -28,6 +28,13 @@ def main():
     return SETTLEMENT_IE({'doc': img_path, 'layout': json.loads(layout)})
 
 
+@app.route('/text', methods=['POST'])
+@process_request
+def text():
+    t = request.form.get('text')
+    return SETTLEMENT_IE(t)
+
+
 if __name__ == '__main__':
     logging.config.dictConfig(LOGGING_CONFIG)
     app.run('0.0.0.0', 5002)
diff --git a/util/model_util.py b/util/model_util.py
index 3d9634e..20ad581 100644
--- a/util/model_util.py
+++ b/util/model_util.py
@@ -38,6 +38,22 @@ def ie_settlement(img_path, layout):
         return None
 
 
+@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
+       after=lambda x: logging.warning('从文本抽取基本医保结算单失败！'))
+def ie_settlement(text):
+    """
+    请求基本医保结算单信息抽取接口
+    :param text: 待抽取文本
+    :return: 抽取结果
+    """
+    url = 'http://ie_settlement:5002/text'
+    response = requests.post(url, {'text': text})
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return None
+
+
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
        after=lambda x: logging.warning('抽取出院记录失败！'))
 def ie_discharge(img_path, layout):
@@ -55,6 +71,22 @@ def ie_discharge(img_path, layout):
         return None
 
 
+@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
+       after=lambda x: logging.warning('从文本抽取出院记录失败！'))
+def ie_discharge(text):
+    """
+    请求出院记录信息抽取接口
+    :param text: 待抽取文本
+    :return: 抽取结果
+    """
+    url = 'http://ie_discharge:5003/text'
+    response = requests.post(url, {'text': text})
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return None
+
+
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
        after=lambda x: logging.warning('抽取费用清单失败！'))
 def ie_cost(img_path, layout):
@@ -72,6 +104,22 @@ def ie_cost(img_path, layout):
         return None
 
 
+@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
+       after=lambda x: logging.warning('从文本抽取费用清单失败！'))
+def ie_cost(text):
+    """
+    请求费用清单信息抽取接口
+    :param text: 待抽取文本
+    :return: 抽取结果
+    """
+    url = 'http://ie_cost:5004/text'
+    response = requests.post(url, {'text': text})
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return None
+
+
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
        after=lambda x: logging.warning('获取图片方向失败！'))
 def clas_orientation(img_path):