Update app.py

app.py

@@ -7,14 +7,13 @@ import streamlit as st
 import numpy as np
 import requests
 from io import BytesIO
-import easyocr as ocr 
 
 def local_css(file_name):
     with open(file_name) as f:
         st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
 
 st.set_page_config(
-   page_title="Streamlit iCodeIdoia - OCR an IMAGE - Extract text from an image",
+   page_title="Streamlit iCodeIdoia",
    page_icon="images/ilpicon1.png",layout="wide",initial_sidebar_state="expanded"
 )
 
@@ -23,62 +22,81 @@ st.image("images/banner.jpg")
 # ---- LOAD
 local_css("styles/style.css")
 
-@st.cache_resource
-def load_model(): 
-    reader = ocr.Reader(['en'],model_storage_directory='.')
-    return reader 
+def fill_square_cropper(img):
+    imgsz = [img.height, img.width]
 
-reader = load_model() #load model
+    original_size = imgsz
+
+    smallestsize = min(imgsz)
+    biggestsize = max(imgsz)
+
+    #get vertical color filler
+    avg_color_per_row = np.average(img, axis=0)
+    avg_color = np.average(avg_color_per_row, axis=0)
+
+    if img.height > img.width:
+      #height is bigger than width
+      area = (0, 0, img.width + (img.height - img.width),img.height)
+      cropped_img = img.crop(area)
+      newimgsz = [cropped_img.height, cropped_img.width]
+    
+      newimg = Image.new('RGB', ([newimgsz[1],newimgsz[0]]), (round(avg_color[0]), round(avg_color[1]), round(avg_color[2])))
+
+      newpos = (img.height-img.width)
+      newpos = newpos/2
+      newimg.paste(img,(int(newpos),0))
+
+      return newimg
+
+    #vertically add color
+    if img.width > img.height: 
+      area = (0, 0, img.width, img.height+ (img.width - img.height))
+      cropped_img = img.crop(area)
+      newimgsz = [cropped_img.height, cropped_img.width]
+    
+      newimg = Image.new('RGB', ([newimgsz[1],newimgsz[0]]), (round(avg_color[0]), round(avg_color[1]), round(avg_color[2])))
+ 
+      newpos = (img.width-img.height)
+      newpos = newpos/2
+      newimg.paste(img,(0,(int(newpos))))
+
+      return newimg
 
 # ---- TABS
 tab1, tab2 = st.tabs(["Demo","Application"])
 
 with tab1:   
    # Handle first image
-   
-   url = "https://https://raw.githubusercontent.com/webdevserv/images_video/main/ocr_sample.jpg" 
+   url = "https://raw.githubusercontent.com/webdevserv/images_video/main/cowportrait.jpg" 
+   # Handle second image
+   url2 = "https://raw.githubusercontent.com/webdevserv/images_video/main/cowlandscape.jpg"
 
-   st.subheader("OCR an image demo")
-   img_description = st.text('Image text will extracted using OCR.')
+   st.subheader("Square an image demo")
+   img_description = st.text('Image will be squared with color filler where applicable.')
 
-   if st.button('OCR Demo'):  
+   if st.button('Square and Fill Demo'):  
     response = requests.get(url)
     img = Image.open(BytesIO(response.content))
-    st.image(input_image) #display image
     img.load()
-    st.image(img) #display image
 
-    with st.spinner("🔄 OCR in process."):
-        result = reader.readtext(np.array(img))
-        result_text = [] #empty list
-        for text in result:
-            result_text.append(text[1])
+    generated_img = fill_square_cropper(img)
+    st.image(generated_img)
 
-        st.write(result_text)
-    st.balloons()
-   else:
-    st.write("Upload an image to extract text using OCR.")
+    response = requests.get(url2)
+    img = Image.open(BytesIO(response.content))
+    img.load()
 
+    generated_img = fill_square_cropper(img)
+    st.image(generated_img)
    
 with tab2:
-  st.subheader("OCR an image app")
-  img_description = st.text('Image text will be extracted using OCR.')
-  uploaded_file = st.file_uploader("Upload a image to OCR.", type=['jpg'])
+  st.subheader("Square an image app")
+  img_description = st.text('Image will be squared with color filler where applicable.')
+  uploaded_file = st.file_uploader("Upload a JPG image to square and fill with color.", type=['jpg'])
 
   if uploaded_file is not None: 
    img = Image.open(uploaded_file)
    img.load()
-   st.image(img) #display image
-
-   with st.spinner("🔄 OCR in process."):
-        result = reader.readtext(np.array(img))
-
-        result_text = [] #empty list for results
-
-        for text in result:
-            result_text.append(text[1])
-
-        st.write(result_text)
-   st.balloons()
-  else:
-   st.write("Upload an image to extract text using OCR.")
+   generated_img = fill_square_cropper(img)
+   st.image(generated_img)
+   st.balloons()