forked from justinyangusa/CodeDay-ify
-
Notifications
You must be signed in to change notification settings - Fork 1
/
app.py
67 lines (56 loc) · 1.82 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
from PIL import Image, ImageChops, ImageEnhance
import requests
import os
from flask import Flask, redirect, jsonify, render_template, request, send_file
import werkzeug
import datetime
import uuid
app = Flask(__name__)
ALLOWED_IMAGE_EXTENSIONS = set(['png', 'jpg', 'jpeg'])
def process_image(img):
#open up the mask
mask = Image.open('mask.png')
mask = mask.convert('RGBA')
#make sure it matches the size of the image
mask = mask.resize(img.size)
#make sure our image has alpha channel
img = img.convert('RGBA')
#unique name
filename = uuid.uuid4().hex + '.png'
filename = os.path.join('/tmp', filename)
Image.alpha_composite(img, mask).save(filename, 'PNG')
#send it back
return filename
@app.route('/')
def index():
return render_template('index.html')
@app.route('/codedayify', 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, mimetype='image/png', as_attachment=True, attachment_filename='spectrafy.png')
if __name__ == '__main__':
port = int(os.environ.get("PORT", 5000))
app.run(host='0.0.0.0', port=port)