HYTOS

from flask import Flask, jsonify, request

import cv2

import numpy as np

import sys, os

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\CRAFT_pytorch_master')

#sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\deep_text_recognition_benchmark_master')

app = Flask(__name__)

@app.route('/')

def index():

    return 'Hello Flask'

@app.route('/text_box', methods=['POST'])

def text_box():

    import text_craft

    r = request

    nparr = np.fromstring(r.data, np.uint8)

    img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)

    #img = img.reshape(1, -1)

    boxes = text_craft.get_text_box(img, img_path=None, score_path=None, trained_model=os.path.dirname(os.path.realpath(__file__)) + '\\CRAFT_pytorch_master\\weights\\craft_mlt_25k.pth')

    return jsonify({'text_box': boxes})

if __name__ == '__main__':

    app.run(debug=True)