RP-Azul commited on
Commit
a0c420a
·
verified ·
1 Parent(s): 2f36540

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +53 -0
app.py ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import easyocr
3
+ import cv2
4
+ import numpy as np
5
+ from PIL import Image
6
+ import tempfile
7
+ import os
8
+
9
+ st.set_page_config(page_title="🧾 EasyOCR Image Text Extractor", layout="centered")
10
+ st.title("🧾 Image OCR with EasyOCR")
11
+
12
+ # Upload image file
13
+ uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
14
+
15
+ def process_image(image_path, gpu=False):
16
+ reader = easyocr.Reader(['en'], gpu=gpu)
17
+ result = reader.readtext(image_path)
18
+
19
+ text_lines = [detection[1] for detection in result]
20
+ saved_text = " ".join(text_lines)
21
+
22
+ img = cv2.imread(image_path)
23
+ for detection in result:
24
+ top_left = tuple([int(val) for val in detection[0][0]])
25
+ bottom_right = tuple([int(val) for val in detection[0][2]])
26
+ text = detection[1]
27
+ font = cv2.FONT_HERSHEY_SIMPLEX
28
+ img = cv2.rectangle(img, top_left, bottom_right, (0, 255, 0), 2)
29
+ img = cv2.putText(img, text, top_left, font, 0.6, (255, 255, 255), 1, cv2.LINE_AA)
30
+
31
+ return img, saved_text
32
+
33
+ if uploaded_file:
34
+ # Save the uploaded file temporarily
35
+ with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
36
+ temp_file.write(uploaded_file.read())
37
+ temp_path = temp_file.name
38
+
39
+ # Process image
40
+ img_result, extracted_text = process_image(temp_path)
41
+
42
+ # Show image with bounding boxes
43
+ st.image(cv2.cvtColor(img_result, cv2.COLOR_BGR2RGB), caption="Detected Text", use_column_width=True)
44
+
45
+ # Show extracted text
46
+ st.subheader("📄 Extracted Text")
47
+ st.text_area("Detected Text", extracted_text, height=200)
48
+
49
+ # Option to download text
50
+ st.download_button("Download Extracted Text", extracted_text, file_name="extracted_text.txt")
51
+
52
+ # Clean up temporary file
53
+ os.remove(temp_path)