constructor()

in source/infer-ocr/infer-ocr.ts [8:90]
75 lines of code
1 McCabe index (conditional complexity)

    constructor(app: cdk.App, id: string, props?: cdk.StackProps) {
        super(app, id, props);
        this.templateOptions.description = `(SO8023-ocr) - AI Solution Kits - Infer OCR. Template version v1.0.0`;

        const modelType = new cdk.CfnParameter(this, "modelType", {
            default: 'Standard',
            type: 'String',
            description: 'OCR model type, please find the model description in deployment guide, default is Standard.',
            allowedValues: [
                'Standard',
                'Lite'
            ]
        });

        const modelVersion = new cdk.CfnParameter(this, "modelVersion", {
            default: 'v1.0.0',
            type: 'String',
            description: 'Pre-trained model version, this parameter works only for testing, please do NOT change the default value.'
        });

        const map = new cdk.CfnMapping(this, 'modelMap', {
            mapping: {
                'Standard': {
                    name: 'advanced',
                    folder: 'advanced'
                },
                'Lite': {
                    name: 'standard',
                    folder: 'standard'
                },
                'Standard-Apache': {
                    name: 'advanced_apache',
                    folder: 'advanced-apache'
                },
                'Lite-Apache': {
                    name: 'standard_apache',
                    folder: 'standard-apache'
                },
            },
        });

        const modelName = map.findInMap(`${modelType.valueAsString}`, 'name');

        const modelFolder =  map.findInMap(`${modelType.valueAsString}`, 'folder');

        const infer_ocr_lambda = new EfsModelLambda(this, 'infer_ocr_lambda', {
            mountPath: '/mnt/lambda',
            objects: [
                `https://aws-gcr-solutions-assets.s3.cn-northwest-1.amazonaws.com.cn/ai-solution-kit/infer-ocr-model/${modelFolder}/${modelVersion.valueAsString}/keys_v1.txt`,
                `https://aws-gcr-solutions-assets.s3.cn-northwest-1.amazonaws.com.cn/ai-solution-kit/infer-ocr-model/${modelFolder}/${modelVersion.valueAsString}/classifier.onnx`,
                `https://aws-gcr-solutions-assets.s3.cn-northwest-1.amazonaws.com.cn/ai-solution-kit/infer-ocr-model/${modelFolder}/${modelVersion.valueAsString}/det_${modelName}.onnx`,
                `https://aws-gcr-solutions-assets.s3.cn-northwest-1.amazonaws.com.cn/ai-solution-kit/infer-ocr-model/${modelFolder}/${modelVersion.valueAsString}/rec_${modelName}.onnx`,
            ],
            handler: 'infer_ocr_app.lambda_handler',
            code: lambda.Code.fromAsset(path.join(__dirname, './src'), {
                bundling: {
                    image: lambda.Runtime.PYTHON_3_8.bundlingImage,
                    command: [
                        'bash', '-c', [
                            `cp -r /asset-input/* /asset-output/`,
                        ].join(' && ')
                    ],
                }
            }),
            runtime: lambda.Runtime.PYTHON_3_8,
            layerPath: path.join(__dirname, '/src'),
            timeout: cdk.Duration.minutes(1),
            memorySize: 10240,
            environment: {
                PYTHONPATH: '/var/runtime:/opt:/opt/python/lib/python3.8/site-packages:python:python/lib/python3.8/site-packages',
                ACCESS_POINT: '/mnt/lambda',
                MODEL_NAME: `${modelName}`,
            }
        });

        const api = new AiKitsRestApi(this, "aikits", {
            apiName: "InferOCRRESTAPI",
            stageName: 'prod',
            resourcePath: 'ocr',
            postFunction: infer_ocr_lambda.function,
        });
        // api.addMethod('POST', efs_lambda.function);
    }