139 lines
4.2 KiB
Python
139 lines
4.2 KiB
Python
import json
|
||
import logging
|
||
|
||
import requests
|
||
from tenacity import retry, stop_after_attempt, wait_random
|
||
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
|
||
after=lambda x: logging.warning('OCR识别失败！'))
|
||
def ocr(img_path):
|
||
"""
|
||
请求图片OCR识别接口
|
||
:param img_path: 待识别图片路径
|
||
:return: 识别结果
|
||
"""
|
||
url = 'http://ocr:5001'
|
||
response = requests.post(url, {'img_path': img_path})
|
||
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_settlement(img_path, layout):
|
||
"""
|
||
请求基本医保结算单信息抽取接口
|
||
:param img_path: 待抽取图片路径
|
||
:param layout: 图片ocr信息
|
||
:return: 抽取结果
|
||
"""
|
||
url = 'http://ie_settlement:5002'
|
||
response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)})
|
||
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):
|
||
"""
|
||
请求出院记录信息抽取接口
|
||
:param img_path: 待抽取图片路径
|
||
:param layout: 图片ocr信息
|
||
:return: 抽取结果
|
||
"""
|
||
url = 'http://ie_discharge:5003'
|
||
response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)})
|
||
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):
|
||
"""
|
||
请求费用清单信息抽取接口
|
||
:param img_path: 待抽取图片路径
|
||
:param layout: 图片ocr信息
|
||
:return: 抽取结果
|
||
"""
|
||
url = 'http://ie_cost:5004'
|
||
response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)})
|
||
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):
|
||
"""
|
||
请求图片方向分类接口
|
||
:param img_path: 待分类图片路径
|
||
:return: 最有可能的两个图片方向
|
||
"""
|
||
url = 'http://clas_orientation:5005'
|
||
response = requests.post(url, {'img_path': img_path})
|
||
if response.status_code == 200:
|
||
return response.json()
|
||
else:
|
||
return ['0', '90']
|
||
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
|
||
after=lambda x: logging.warning('获取文档区域失败！'))
|
||
def det_book(img_path):
|
||
"""
|
||
请求文档区域识别接口
|
||
:param img_path: 待识别图片路径
|
||
:return: 文档图片路径列表
|
||
"""
|
||
url = 'http://det_book:5006'
|
||
response = requests.post(url, {'img_path': img_path})
|
||
if response.status_code == 200:
|
||
return response.json()
|
||
else:
|
||
return [img_path]
|
||
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
|
||
after=lambda x: logging.warning('矫正扭曲图片失败！'))
|
||
def dewarp(img_path):
|
||
"""
|
||
请求矫正图片接口
|
||
:param img_path: 待矫正图片路径
|
||
:return: 矫正后的图片路径
|
||
"""
|
||
url = 'http://dewarp:5007'
|
||
response = requests.post(url, {'img_path': img_path})
|
||
if response.status_code == 200:
|
||
return response.json()
|
||
else:
|
||
return img_path
|
||
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
|
||
after=lambda x: logging.warning('文本分类失败！'))
|
||
def clas_text(text):
|
||
"""
|
||
请求文本分类接口
|
||
:param text: 待分类文本
|
||
:return: 分类结果
|
||
"""
|
||
if not text:
|
||
return None
|
||
url = 'http://clas_text:5008'
|
||
response = requests.post(url, {'text': text})
|
||
if response.status_code == 200:
|
||
return response.json()
|
||
else:
|
||
return None
|