import logging.config
import os
import traceback

os.environ["CUDA_VISIBLE_DEVICES"] = "1"

from auto_email.error_email import send_an_error_email
from config.log import LOGGING_CONFIG
from photo_review.photo_review import main

# 项目必须从此处启动，否则代码中的相对路径可能导致错误的发生
if __name__ == '__main__':
    logging.config.dictConfig(LOGGING_CONFIG)
    log = logging.getLogger()

    # 崩溃后的重试次数
    for _ in range(2):
        try:
            log.info("【照片审核关键信息抽取】开始")
            main()
        except Exception as e:
            log.error(traceback.format_exc())
            send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc())