RP-Azul commited on
Commit
bec5916
·
verified ·
1 Parent(s): e2470cc

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -53
app.py DELETED
@@ -1,53 +0,0 @@
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)