# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: MIT-0
"""Top-level entrypoint script for alternative OCR engines
"""
# Python Built-Ins:
import logging
import os
import sys


def run_main():
    """Configure logging, import local modules and run the job"""
    consolehandler = logging.StreamHandler(sys.stdout)
    consolehandler.setFormatter(
        logging.Formatter("%(asctime)s [%(name)s] %(levelname)s %(message)s")
    )
    logging.basicConfig(handlers=[consolehandler], level=os.environ.get("LOG_LEVEL", logging.INFO))

    from textract_transformers.ocr import main

    return main()


if __name__ == "__main__":
    # If the file is running as a script, we're in batch processing mode and should run the batch
    # routine (with a little logging setup before any imports, to make sure output shows up ok):
    run_main()
else:
    # If the file is imported as a module, we're in inference mode and should pass through the
    # override functions defined in the inference module.
    from textract_transformers.ocr import *