import config.config
import application_extractor.ocr_PP_StructureV3
import application_extractor.rectify_OCR_result

multi_imgs = [
    "E:\\project\\arbitration_system\\application_extractor\\test\\刘正新\\刘正新-申请书.png",
    "E:\\project\\arbitration_system\\application_extractor\\test\\刘正新\\刘正新-申请书续.png"
    # "E:\\project\\arbitration_system\\appplication_extractor\\test\\李述花\\李述花-申请书.png"
]

def application_extractor_run(multi_imgs):

    application_client_ocr = application_extractor.ocr_PP_StructureV3.LayoutParserClient_application()
    result_ocr = application_client_ocr.parse(multi_imgs)

    application_client_rectify = application_extractor.rectify_OCR_result.RectifyClient_application()
    result_rectify = application_client_rectify.extract_legal_document(result_ocr)


    return result_rectify

if __name__ == '__main__':
    print(application_extractor_run(multi_imgs))