FacialRecognition.agent.nut

// MIT License
// Copyright 2017 Electric Imp
// SPDX-License-Identifier: MIT
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO
// EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES
// OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.

// ---------------- Kairos attempt ---------------------

appId <- "<YOUR APP ID>";
appKey <- "<YOUR API KEY>";

galleryName <- "ElectricImp";
subjectName <- "<YOUR NAME>";

function enroll(image, name, gallery) {
    local headers = {
        "Content-Type" : "application/json",
        "app_id" : appId,
        "app_key" : appKey
    }
    local parameters = http.jsonencode({
        "image" : http.base64encode(image),
        "subject_id" : name,
        "gallery_name" : gallery
    });
    local send_url = "https://api.kairos.com/enroll";
    local request = http.post(send_url, headers, parameters);
    local response = request.sendsync();
}

function deleteGallery(gallery) {
    local headers = {
        "Content-Type" : "application/json",
        "app_id" : appId,
        "app_key" : appKey
    };
    
    local url = "https://api.kairos.com/gallery/remove";
    local data = http.jsonencode({
        "gallery_name" : gallery
    });
    local request = http.post(url, headers, data);
    local response = http.jsondecode(request.sendsync());
    if ("status" in response && response.status == "Complete") {
        server.log("successfully deleted gallery");
    }
}

function detect(img) {
    local headers = {
        "Content-Type" : "application/json",
        "app_id" : appId,
        "app_key" : appKey
    };
    
    local url = "https://api.kairos.com/recognize";
    local data = http.jsonencode({
        "image" : http.base64encode(img),
        "gallery_name" : galleryName
    });
    
    local request = http.post(url, headers, data);
    local response = request.sendsync();
    local pp = PrettyPrinter();
    server.log(pp.format(http.jsondecode(response.body)));
    local ret = http.jsondecode(response.body);
    if ("images" in ret) {
        server.log("image found");
        if ("transaction" in ret.images[0] && ret.images[0].transaction.status == "success") {
            server.log("success");
            local myCandidates = ret.images[0].candidates;
            if (myCandidates.len() > 0) {
                local topCandidate = myCandidates[0].subject_id;
                local maxVal = myCandidates[0].confidence;
                
                foreach (i in myCandidates) {
                    if (i.confidence > maxVal) {
                        topCandidate = i.subject_id;   
                        maxVal = i.confidence;
                    }
                }
                
                if (maxVal > 0.5) {
                    local ret = format("Hello %s",
                    topCandidate, maxVal);
                    server.log(ret);
                    // Do something with topCandidate here...
                }
            }
            
        }
    } else {
        // Uh-oh, something went wrong
    }
     
    device.send("done", "");
}

device.on("detect", function(img) {
    detect(img);
});

device.on("enroll", function(img) {
    enroll(img, subjectName, galleryName);
});

device.on("something" function(v) {
    server.log("Something came into the frame!");
});