Update app.py

app.py

@@ -4,69 +4,73 @@ import streamlit as st
 import easyocr
 import PIL
 from PIL import Image, ImageDraw
+from streamlit_drawable_canvas import st_canvas
 
 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 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
+        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
+    # Display image on streamlit
     st.image(image)
 
-
-# main title
+# Main title
 st.title("Get text from image with EasyOCR")
 
-# subtitle
+# Subtitle
 st.markdown("## EasyOCR with Streamlit")
 
-# upload image file
-file = st.file_uploader(label = "Upload Here", type=['png', 'jpg', 'jpeg'])
+# Upload image file or draw
+st.markdown("## Upload an Image or Draw")
+col1, col2 = st.columns(2)
+
+with col1:
+    file = st.file_uploader("Upload Here", type=['png', 'jpg', 'jpeg'])
 
-#read the csv file and display the dataframe
+with col2:
+    # Drawable canvas
+    canvas_result = st_canvas(
+        fill_color="rgba(255, 165, 0, 0.3)", 
+        stroke_width=3,
+        stroke_color="#ffffff",
+        background_color="#000000",
+        background_image=None if file else st.session_state.get("background", None),
+        update_streamlit=True,
+        width=400,
+        height=400,
+        drawing_mode="freedraw",
+        key="canvas",
+    )
+
+# Process uploaded image or drawing
 if file is not None:
-    image = Image.open(file) # read image with PIL library
-    st.image(image) #display
+    image = Image.open(file)  # Read image with PIL library
+elif canvas_result.image_data is not None:
+    image = Image.fromarray(canvas_result.image_data.astype('uint8'), 'RGBA').convert('RGB')
+else:
+    st.write("Please upload an image or use the canvas to draw.")
+    image = None
 
-    # it will only detect the English and Turkish part of the image as text
-    reader = easyocr.Reader(['my','en'], gpu=False) 
-    result = reader.readtext(np.array(image))  # turn image to numpy array
+if image is not None:
+    st.image(image)  # Display
 
-    # Add a placeholder
-    # latest_iteration = st.empty()
-    # bar = st.progress(0)
-    
-    # for i in range(100):
-      # Update the progress bar with each iteration.
-      # latest_iteration.text(f'Iteration {i+1}')
-      # bar.progress(i + 1)
-      # time.sleep(0.1)
+    # Only detect the English and Turkish part of the image as text
+    reader = easyocr.Reader(['en','my'], gpu=False) 
+    result = reader.readtext(np.array(image))  # Turn image to numpy array
 
-    # print all predicted text:
-    for idx in range(len(result)):
-        pred_text = result[idx][1]
+    # Print all predicted text:
+    for idx, res in enumerate(result):
+        pred_text = res[1]
         st.write(pred_text)
     
-    # collect the results in the dictionary:
-    textdic_easyocr = {}
-    for idx in range(len(result)):
-        pred_coor = result[idx][0]
-        pred_text = result[idx][1]
-        pred_confidence = result[idx][2]
-        textdic_easyocr[pred_text] = {}
-        textdic_easyocr[pred_text]['pred_confidence'] = pred_confidence
+    # Collect the results in the dictionary:
+    textdic_easyocr = {res[1]: {'pred_confidence': res[2]} for res in result}
 
-    # create a data frame which shows the predicted text and prediction confidence
+    # Create a data frame which shows the predicted text and prediction confidence
     df = pd.DataFrame.from_dict(textdic_easyocr).T
     st.table(df)
 
-    # get boxes on the image 
+    # Get boxes on the image 
     rectangle(image, result)
-
-    st.spinner(text="In progress...")
-    
-else:
-    st.write("Upload your image")