import argparse
from google.cloud import vision
from google.cloud.vision import types
import io
from gcloud import storage

def create_uri(bucket_name, blob_name):
    return "gs://" + bucket_name + "/" + blob_name

def get_text(image_uri):
    client = vision.ImageAnnotatorClient()
    image = vision.types.Image()
    image.source.image_uri = image_uri
    response = client.document_text_detection(image=image)
    document = response
    return response

def get_all_text(bucket_name, directory):
    client = storage.Client(project='medical-extraction')
    bucket = client.get_bucket(bucket_name)
    full_text = []

    for blob in bucket.list_blobs(prefix=directory):
        print("----------------------" + blob.name + "---------------------------")
        next_doc = get_text(create_uri(bucket_name,blob.name)).full_text_annotation.text.splitlines()
        full_text += next_doc
        # print(full_text)
        for line in next_doc:
            print (line)

    return full_text