Update app.py

app.py

@@ -5,6 +5,18 @@ import easyocr
 import PIL
 from PIL import Image
 
+def rectangle(image, result):
+    # https://www.blog.pythonlibrary.org/2021/02/23/drawing-shapes-on-images-with-python-and-pillow/
+    """ draw rectangles on image based on predicted coordinates"""
+    draw = ImageDraw.Draw(image)
+    for res in result:
+        top_left = tuple(res[0][0]) # top left coordinates as tuple
+        bottom_right = tuple(res[0][2]) # bottom right coordinates as tuple
+        draw.rectangle((top_left, bottom_right), outline="blue", width=2)
+    #display image on streamlit
+    st.image(image)
+
+
 # main title
 st.title("Get text from image with EasyOCR")
 
@@ -12,7 +24,7 @@ st.title("Get text from image with EasyOCR")
 st.markdown("## EasyOCRR with Streamlit")
 
 # upload image file
-file = st.file_uploader(label = "Upload your image", type=['png', 'jpg', 'jpeg'])
+file = st.file_uploader(label = "Upload Here", type=['png', 'jpg', 'jpeg'])
 
 #read the csv file and display the dataframe
 if file is not None:
@@ -35,6 +47,9 @@ if file is not None:
     # create a dataframe which shows the predicted text and prediction confidence
     df = pd.DataFrame.from_dict(textdic_easyocr).T
     st.table(df)
+
+    # get boxes on the image 
+    rectangle(image, result)
     
 else:
     st.write("Upload your image")