forked from AKPsi-OT/akpsi_rush_filter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
140 lines (113 loc) · 4.25 KB
/
app.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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
from PIL import Image, ImageChops, ImageEnhance
import requests
import os
from flask import Flask, redirect, jsonify, render_template, request, send_file, send_from_directory, url_for, session
import werkzeug
import datetime
import uuid
from flask_oauthlib.client import OAuth, OAuthException
# Binary response content parsing
from PIL import Image
from io import BytesIO
FACEBOOK_APP_ID = '1813961962211551'
FACEBOOK_APP_SECRET = '1d4efe2619879a1cd22e39e30767e4fb'
app = Flask(__name__)
app.debug = True
app.secret_key = 'development'
oauth = OAuth(app)
facebook = oauth.remote_app(
'facebook',
consumer_key=FACEBOOK_APP_ID,
consumer_secret=FACEBOOK_APP_SECRET,
request_token_params={'scope': 'email'},
base_url='https://graph.facebook.com',
request_token_url=None,
access_token_url='/oauth/access_token',
access_token_method='GET',
authorize_url='https://www.facebook.com/dialog/oauth'
)
ALLOWED_IMAGE_EXTENSIONS = set(['png', 'jpg', 'jpeg'])
def process_image(img):
#open up the mask
mask = Image.open('static/images/mask.png')
mask = mask.convert('RGBA')
mask = mask.resize(img.size)
img.paste(mask, (0,0), mask)
newdata = img.getdata()
#create an image from our new combined data
img.putdata(newdata)
#unique name
filename = uuid.uuid4().hex + '.png'
filename = os.path.join('/tmp', filename)
img.save(filename, 'PNG')
#send it back
return filename
@app.route('/')
def index():
return render_template('index.html')
@app.route('/questify', methods=['POST'])
def classify_upload():
try:
#get the image from the request
imagefile = request.files['imagefile']
filename_ = str(datetime.datetime.now()).replace(' ', '_') + \
werkzeug.secure_filename(imagefile.filename)
filename = os.path.join('/tmp', filename_)
#make sure it has the correct file type
b = False
for ext in ALLOWED_IMAGE_EXTENSIONS:
if ext in filename:
b = True
if not b:
return 'Invalid filetype.'
#save the file to /tmp
imagefile.save(filename)
#open the image for Pillow
image = Image.open(filename)
except Exception as err:
#uh oh. Something went wrong.
print ('Uploaded image open error: ' + err)
return 'Error: ' + err
#process the image
resultFilename = process_image(image)
#send it back
# return send_file(resultFilename)
return render_template('questified.html', final_image=resultFilename[5:], desc="Interested in joining a multidisciplinary, hands-on honors program focused on intellectual curiosity, critical thinking, professionalism and teamwork? Apply to QUEST at: apply.questumd.com!")
@app.route('/tmp/<filename>')
def send_file(filename):
return send_from_directory('/tmp', filename)
@app.route('/login')
def login():
callback = url_for(
'facebook_authorized',
next=request.args.get('next') or request.referrer or None,
_external=True
)
return facebook.authorize(callback=callback)
@app.route('/login/authorized')
def facebook_authorized():
resp = facebook.authorized_response()
if resp is None:
return 'Access denied: reason=%s error=%s' % (
request.args['error_reason'],
request.args['error_description']
)
if isinstance(resp, OAuthException):
return 'Access denied: %s' % resp.message
session['oauth_token'] = (resp['access_token'], '')
me = facebook.get('/me')
# return 'Logged in as id=%s name=%s redirect=%s' % \
# (me.data['id'], me.data['name'], request.args.get('next'))
payload = {'type':'large','redirect':'true','width':'500','height':'500'}
r = requests.get("http://graph.facebook.com/" + me.data['id'] + "/picture", params=payload)
img = Image.open(BytesIO(r.content))
#process the image
resultFilename = process_image(img)
#send it back
return render_template('questified.html', final_image=resultFilename[5:], desc="Interested in joining a multidisciplinary, hands-on honors program focused on intellectual curiosity, critical thinking, professionalism and teamwork? Apply to QUEST at: apply.questumd.com!")
@facebook.tokengetter
def get_facebook_oauth_token():
return session.get('oauth_token')
if __name__ == '__main__':
port = int(os.environ.get('PORT', 5000))
app.run(host='0.0.0.0', port=port)