BIOML commited on
Commit
28b349c
·
1 Parent(s): 64cfb47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -47
app.py CHANGED
@@ -1,47 +1,31 @@
1
- import easyocr as ocr #OCR
2
- import streamlit as st #Web App
3
- from PIL import Image #Image Processing
4
- import numpy as np #Image Processing
5
-
6
- #title
7
- st.title("Easy OCR - Extract Text from Images")
8
-
9
- #subtitle
10
- st.markdown("## Optical Character Recognition - Using `easyocr`, `streamlit` - hosted on 🤗 Spaces")
11
-
12
- st.markdown("Link to the app - [image-to-text-app on 🤗 Spaces](https://huggingface.co/spaces/Amrrs/image-to-text-app)")
13
-
14
- #image uploader
15
- image = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpeg'])
16
-
17
-
18
- @st.cache
19
- def load_model():
20
- reader = ocr.Reader(['en'],model_storage_directory='.')
21
- return reader
22
-
23
- reader = load_model() #load model
24
-
25
- if image is not None:
26
-
27
- input_image = Image.open(image) #read image
28
- st.image(input_image) #display image
29
-
30
- with st.spinner("🤖 AI is at Work! "):
31
-
32
-
33
- result = reader.readtext(np.array(input_image))
34
-
35
- result_text = [] #empty list for results
36
-
37
-
38
- for text in result:
39
- result_text.append(text[1])
40
-
41
- st.write(result_text)
42
- #st.success("Here you go!")
43
- st.balloons()
44
- else:
45
- st.write("Upload an Image")
46
-
47
- st.caption("Made with ❤️ by @1littlecoder. Credits to 🤗 Spaces for Hosting this ")
 
1
+ from analze import *
2
+
3
+ app = Flask(__name__)
4
+
5
+ @app.route('/')
6
+ def home():
7
+ return render_template('home.html')
8
+
9
+ @app.route('/upload', methods=['GET', 'POST'])
10
+ def upload_file():
11
+ if request.method == 'POST':
12
+ # Check if a file was uploaded
13
+ if 'file' not in request.files:
14
+ return render_template('home.html', content='No file uploaded.')
15
+ file = request.files['file']
16
+ # Check if the file has a filename
17
+ if file.filename == '':
18
+ return render_template('home.html', content='No file selected.')
19
+ filepath = 'email files/' + file.filename
20
+ return render_template('home.html',
21
+ content=check_file_type(file),
22
+ features = get_features(filepath),
23
+ pre_content=predict_content(text_feature(filepath)),
24
+ pre_tag=predict_html(html_tags_feature(filepath)),
25
+ pre_num=predict_num(num_feature(filepath)),
26
+ pre_extra=predict_extra(extra_feature(filepath)))
27
+
28
+ return render_template('home.html')
29
+
30
+ if __name__ == '__main__':
31
+ app.run(host='127.0.0.1',port=5000)