websocket ocr server

ocr-server.py

#!/usr/bin/python
# coding:utf-8
from websocket_server import WebsocketServer
import pytesseract
import io
import requests
from PIL import Image


def get_ocr_result(img_url):
    content = requests.get(img_url).content
    img = Image.open(io.BytesIO(content))
    text = pytesseract.image_to_string(img)
    return text


def message_received(client, server, msg):
    try:
        cmd, data = msg.split(' ')
        if cmd == 'ocr':
            img_url = data
            ocr_result = get_ocr_result(img_url)
            server.send_message(client, 'ocr_result '+ocr_result)
    except Exception as e:
        print(e)


def main():
    port = 9001
    server = WebsocketServer(port)
    server.set_fn_message_received(message_received)
    server.run_forever()


if __name__ == '__main__':
    main()