import easyocr as ocr  #OCR
from PIL import Image #Image Processing
import numpy as np #Image Processing 
import gradio as gr


input = gr.inputs.Image(label="Upload your Image", type = type=['png','jpg','jpeg'], optional=True)

def load_model(): 
    reader = ocr.Reader(['en'],model_storage_directory='.')
    return reader 

def text(image):
    reader = load_model() #load model
    input_image = Image.open(image) #read image
    result = reader.readtext(np.array(input_image))
    result_text = [] #empty list for results
    for text in result:
        result_text.append(text[1])
    return result_text

output = gr.outputs.Textbox(type="text",label="Captions")

title = "Image Captioning "

interface = gr.Interface(
        fn=text,
        inputs = input,
        theme="grass",
        outputs=output,
        title=title,
    )
interface.launch(debug=True)