Client.py

############################################################################################
#
# Project:       Peter Moss Acute Myeloid & Lymphoblastic Leukemia AI Research Project
# Repository:    ALL Detection System 2019
# Project:       Facial Authentication Server
#
# Author:        Adam Milton-Barker (AdamMiltonBarker.com)
# Contributors:
# Title:         Client Class
# Description:   Sends single or multiple images to the ALL Detection System 2019 Classifier.
# License:       MIT License
# Last Modified: 2020-07-21
#
############################################################################################

import cv2, json, os, requests, sys, time

from Classes.Helpers import Helpers


class Client():
    """ ALL Detection System 2019 Client Class

    Sends single or multiple images to the ALL Detection System 2019 Classifier.
    """

    def __init__(self):
        """ Initializes the Client Class. """

        self.Helpers = Helpers("Client")
        self.confs = self.Helpers.confs

        self.addr = "http://"+self.confs["Classifier"]["IP"] + \
            ':'+str(self.confs["Classifier"]["Port"]) + '/Inference'
        self.headers = {'content-type': 'image/jpeg'}

        self.Helpers.logger.info("Classifier class initialization complete.")

    def send(self, imagePath):
        """ Sends image to the inference API endpoint. """

        _, img_encoded = cv2.imencode('.png', cv2.imread(imagePath))
        response = requests.post(
            self.addr, data=img_encoded.tostring(), headers=self.headers)
        response = json.loads(response.text)

        self.Helpers.logger.info(imagePath + ": " + response["Message"])

    def test(self):
        """ Loops through all images in the testing directory and sends
        them to the inference API endpoint. """

        testingDir = self.confs["Classifier"]["NetworkPath"] + \
            self.confs["Classifier"]["TestImagePath"] + "/"

        for test in os.listdir(testingDir):
            if os.path.splitext(test)[1] in self.confs["Classifier"]["ValidIType"]:
                self.send(testingDir+test)
                time.sleep(7)


if __name__ == "__main__":

    Client = Client()

    if sys.argv[1] == "Send":

        Client.send(sys.argv[2])

    elif sys.argv[1] == "Test":

        Client.test()