百度云接口应用示例

pip install pillow

​

import datetime

import requests

import warnings

from flask import Flask,request

import json

import base64

from PIL import Image

from io import BytesIO

​

warnings.filterwarnings("ignore")

​

# 百度云上应用的API Key

client_id = "xxxxx"

​

# 百度云上应用的Secret Key

client_secret = "xxxxx"

​

​

# 定义一个类，用来每2小时更新access_token

class setup:

​

    def __init__(self):

        self.t0 = ''

        self.gettoken()

​

    def timecheck(self):

        self.t1 = datetime.datetime.now()

        self.total_interval_time = (self.t1 - self.t0).total_seconds()

        if self.total_interval_time > 2592000:

            self.gettoken()

​

    def gettoken(self):

        self.url = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + client_id + '&client_secret=' + client_secret

        self.token = requests.get(self.url, data='', headers='', verify=False)

        self.token = self.token.json()

        self.token = self.token['access_token']

        self.t1 = datetime.datetime.now()

        self.t0 = self.t1

​

    def s(self):

        self.timecheck()

        return self.token

​

# 定义一个函数，将简道云传过来的URL格式图片转为Base64编码

def download_img(img_url):

    header={}

    r = requests.get(img_url, headers=header, stream=True)

    image = Image.open(BytesIO(r.content)) 

    img_64 = base64.b64encode(BytesIO(r.content).read())

    return img_64

​

# 定义一个函数，调用百度云的通用文字识别（标准版），来获取数据

def img_ocr(img_url):

    request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic"

    img = download_img(img_url)

    params = {"image":img}

    access_token = token_

    request_url = request_url + "?access_token=" + access_token

    headers = {'content-type': 'application/x-www-form-urlencoded'}

    response = requests.post(request_url, data=params, headers=headers)

    if response:

        return response.json()

​

​

# 实例化Flask类

app = Flask(__name__)

​

# 定义一个接口，用来和简道云前端事件进行交互

# 简道云表单上的图片控件发送值变动，触发事件传入数据至接口

# 简道云传入 {"image_url":xxx}

# 接口返回 {"words_result":xxx}

​

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

def hello_world():

    image_url = json.loads(request.data).get('image_url')

    if image_url != '':

        ocr_data = img_ocr(image_url)

        words_result = ""

        #由于百度云识别的隔行数据分布在不同的数组里面，我们稍微处理下

        for each in ocr_data['words_result']:

            words_result = words_result + each['words'] + '\n'

        return json.dumps({"words_result":words_result})

    else:

        return json.dumps({"words_result":"无内容"})

​

​

if __name__ == '__main__':

    # 程序开始时，实例化我们定义的类

    init = setup()

    # 每2小时更新access_token

    token_ = init.s()

    app.run(host='0.0.0.0',port=10000)

​