import easyocr as ocr #OCR
import streamlit as st #Web App
from PIL import Image #Image Processing
import numpy as np #Image Processing

#title
st.title("Easy OCR - Extract Text from Images")

#subtitle
st.markdown("## Using `easyocr`, `streamlit`")

#image uploader
image = st.file_uploader(label = "Upload your image here", type=['png', 'jpg', 'jpeg'])

# @st.cache
# def load_model():
#     reader = ocr.Reader(['en'])
#     return reader
#
# reader = load_model() #load model

if image is not None:
    input_image = Image.open(image) #read image
    st.image(input_image) #display image

    with st.spinner("🤖 AI is at Work! "):
        reader = ocr.Reader(['en'])
        result = reader.readtext(np.array(input_image))

        result_text = []

        for text in result:
            result_text.append(text[1])

        st.write(result_text)
    st.success("Here you go!")
else:
    st.write("Upload an Image")

st.caption("Made with ❤️ by z")