-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain_upload.py
94 lines (71 loc) · 2.82 KB
/
main_upload.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# -*- coding: utf-8 -*-
import os
import time
import hashlib
import requests
import pytesseract
from pytesseract import image_to_string
from flask import Flask, render_template, redirect, url_for, request
from flask_uploads import UploadSet, configure_uploads, IMAGES, patch_request_class
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired, FileAllowed
from wtforms import SubmitField
from PIL import Image
from io import BytesIO
from google.cloud import vision
#pytesseract.pytesseract.tesseract_cmd = r'/home/sonwillogistics/env/lib/python3.5/site-packages/pytesseract/pytesseract.py'
app = Flask(__name__)
app.config['SECRET_KEY'] = 'I have a dream'
app.config['UPLOADED_PHOTOS_DEST'] = os.getcwd() + '/static'
photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)
patch_request_class(app) # set maximum file size, default is 16MB
class UploadForm(FlaskForm):
photo = FileField(validators=[FileAllowed(photos, u'Image Only!'), FileRequired(u'Choose a file!')])
submit = SubmitField(u'Upload')
@app.route('/', methods=['GET', 'POST'])
def upload_file():
form = UploadForm()
if form.validate_on_submit():
for filename in request.files.getlist('photo'):
name = ('file' + str(time.time()))[:15]
photos.save(filename, name=name + '.')
success = True
else:
success = False
return render_template('index.html', form=form, success=success)
@app.route('/scan/<filename>')
def scan_file(filename):
file_url = photos.url(filename)
response = requests.get(file_url)
"""Detects text in the file."""
client = vision.ImageAnnotatorClient()
with io.open(path, 'rb') as image_file:
content = image_file.read()
image = vision.types.Image(content=content)
response = client.text_detection(image=image)
texts = response.text_annotations
for text in texts:
print('\n"{}"'.format(text.description))
vertices = (['({},{})'.format(vertex.x, vertex.y)
for vertex in text.bounding_poly.vertices])
print('bounds: {}'.format(','.join(vertices)))
render_template('index.html', file_text = texts)
@app.route('/manage')
def manage_file():
files_list = os.listdir(app.config['UPLOADED_PHOTOS_DEST'])
return render_template('manage.html', files_list=files_list)
@app.route('/open/<filename>')
def open_file(filename):
file_url = photos.url(filename)
response = requests.get(file_url)
img = Image.open(BytesIO(response.content))
file_text = image_to_string(img)
return render_template('browser.html', file_url=file_url, file_text=file_text)
@app.route('/delete/<filename>')
def delete_file(filename):
file_path = photos.path(filename)
os.remove(file_path)
return redirect(url_for('manage_file'))
if __name__ == '__main__':
app.run()