import base64

import cv2
import numpy as np
from sanic import Sanic
from sanic.response import json
from yolov5_onnx_demo.model.yolov5_infer import *


def base64_to_bgr(bs64):
    img_data = base64.b64decode(bs64)
    img_arr = np.fromstring(img_data, np.uint8)
    img_np = cv2.imdecode(img_arr, cv2.IMREAD_COLOR)
    return img_np


app = Sanic('tamper_det')


@app.post('/tamper_det')
def hello(request):
    d = request.json
    print(d['img'])
    img = base64_to_bgr(d['img'])
    result = grpc_detect(img)

    return json({'results': result})


if __name__ == '__main__':
    app.run(host='192.168.10.11', port=8009,workers=10)