cloudvisreq.py

from __future__ import print_function
from base64 import b64encode
from os import makedirs, remove
from os.path import join, basename
from sys import argv
import json
import requests
import glob


ENDPOINT_URL = 'https://vision.googleapis.com/v1/images:annotate'
RESULTS_DIR = 'jsons'

def make_image_data_list(image_filenames):
    """
    image_filenames is a list of filename strings
    Returns a list of dicts formatted as the Vision API
        needs them to be
    """
    img_requests = []
    with open(image_filenames, 'rb') as f:
        ctxt = b64encode(f.read()).decode()
        img_requests.append({
                'image': {'content': ctxt},
                'features': [{
                    'type': 'TEXT_DETECTION',
                    'maxResults': 1
                }]
        })
    return img_requests

def make_image_data(image_filenames):
    """Returns the image data lists as bytes"""
    imgdict = make_image_data_list(image_filenames)
    return json.dumps({"requests": imgdict }).encode()


def request_ocr(api_key, image_filenames):
    response = requests.post(ENDPOINT_URL, data=make_image_data(image_filenames), params={'key': api_key}, headers={'Content-Type': 'application/json'})
    return response


if __name__ == '__main__':
    api_key = "AIzaSyCSMpzBIKlZObk8Uzkx6Iavo967m7vFf0Q"
    file = open("output/words/normal/detection.txt","w")
    list = glob.glob("/Users/Piyush_Jena/Documents/Opensoft/spell_correction/test_software/output/words/normal/*.jpg")
    for i in range(0,len(list)):
        image_filenames = list[i]
        if not api_key or not image_filenames:
            print("""Please supply an api key, then one or more image filenames $ python cloudvisreq.py image1.jpg image2.png""")
        else:
            response = request_ocr(api_key, image_filenames)
            if response.status_code != 200 or response.json().get('error'):
                print(response.text)
            else:
                for idx, resp in enumerate(response.json()['responses']):
                    # save to JSON file
                    imgname = image_filenames[idx]
                    jpath = join(RESULTS_DIR, basename(imgname) + '.json')
                    with open(jpath, 'w') as f:
                        datatxt = json.dumps(resp, indent=2)
                        print("Wrote", len(datatxt), "bytes to", jpath)
                        f.write(datatxt)

                    # print the plaintext to screen for convenience
                    print("---------------------------------------------")
                    t = resp['textAnnotations'][0]
                    print("    Bounding Polygon:")
                    print(t['boundingPoly'])
                    print("    Text:")
                    print(t['description'])

                    file.write(t['description'])
                    #print("---------------------------------------------")

                    #print("    Bounding Polygon:")
                    #print(t['boundingPoly'])
                    #print("    Text:")
                    #print(t['description'])
    for file in list:
        remove(file)