ESP32 Forum

Code: Select all

const AWS = require('aws-sdk');
var rekognition = new AWS.Rekognition({apiVersion: '2016-06-27'});
var s3 = new AWS.S3();
var S3params;
var filePath;
var recoType = "text";

exports.handler = (event, context, callback) => {
    console.log(event.queryStringParameters);
    recoType = event.queryStringParameters.reco;
     let encodedImage = event.body;//JSON.parse(event.body).image;
     let decodedImage = Buffer.from(encodedImage, 'base64');
     filePath = "images/image.jpg";
     S3params = {
       "Body": decodedImage,
       "Bucket": "bucket_name",
       "Key": filePath  
    };
    s3.upload(S3params, function(err, data){
        if(err) {
            console.log("problem with saving to S3");
            callback(err, null);
        }
        else {
    
            var recoParams = {
              Image: { /* required */
                Bytes: decodedImage /* Strings will be Base-64 encoded on your behalf */,
              }
            };
            console.log(recoType);
            if(recoType == "text") {
                rekognition.detectText(recoParams, function(err, data) {
                    if (err) console.log(err, err.stack); // an error occurred
                    else     console.log(data);           // successful response
                    
                    if(err) {
                        callback(err, null);
                    } 
                    else {
                        let response = {
                            "statusCode": 200,
                            // "headers": {
                            // },
                            "body": JSON.stringify(data),
                            "isBase64Encoded": false
                        };
                        callback(null, response);
                    }
                });
            }
            else {
                rekognition.detectLabels(recoParams, function(err, data) {
                    if (err) console.log(err, err.stack); // an error occurred
                    else     console.log(data);           // successful response
                    
                    if(err) {
                        callback(err, null);
                    } 
                    else {
                        let response = {
                            "statusCode": 200,
                            // "headers": {
                            // },
                            "body": JSON.stringify(data),
                            "isBase64Encoded": false
                        };
                        callback(null, response);
                    }
                });
            }
        }
    });
};