[get_score_handler_v2] #python #handler #pytorch

get_score_handler_v2.py

class GetScoreHandler_v2(BaseHandler):

    def __init__(self, *args, **kwargs):
        super(GetScoreHandler_v2, self).__init__(*args, **kwargs)
        self.model = CleanscoringModelV2API()

    @gen.coroutine
    def post(self, slug=None):
        try:
            request_json_data = json.loads(self.request.body)

            results = yield self.get_score(request_json_data)
            self.write(json.dumps(results))
            self.finish()
        except Exception as ex:
            self.handle_exception(ex)

    @run_on_executor
    def get_score(self, request_json_data):
        try:
            request_dict = self.model.daemon_utils.convert(request_json_data)
            prod_no = request_dict.get('prod_no', None)
            image_url = request_dict.get('image_url', None)
            pil_image, img_size = self.model.daemon_utils.pil_image_load(image_url)
            app_log.info(image_url)
            cv2_image, _ = self.model.daemon_utils.image_load_and_resize(image_url, resize=False)
            result = self.model.inference(pil_image, cv2_image)
            app_log.info("image_url:" + request_dict['image_url'] + ", score:" + str(result))

            return self.model.daemon_utils.encode_result_to_json_v2_api([result, img_size, prod_no])
        except Exception as ex:

            self.handle_exception(ex)