Update ocr_server.py

magic_pdf/dict2md/ocr_server.py

@@ -26,6 +26,8 @@ REVISION = 'v1.0.4'
 BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>"
 PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏."
 
+logger.add("../../parse.log", rotation="10 MB", level="INFO",
+           format="{time} {level} {message}", encoding='utf-8', enqueue=True)
 
 def _get_args():
     parser = ArgumentParser()
@@ -253,10 +255,8 @@ if __name__ == "__main__":
     args = _get_args()
     config = configparser.ConfigParser()
     config.read(args.config_path)
-    ocr_server = config.get('server', 'ocr_server')
-    if 'http' in ocr_server:
-        ocr_server = ocr_server.split('://')[1]
-    host,port = ocr_server.split(':')[0],int(ocr_server.split(':')[1])
+    host = config.get('server', 'ocr_host')
+    port = int(config.get('server', 'ocr_port'))
     uvicorn.run(app, host=host, port=port)