def index():
if request.method == 'POST':
file = request.files.get('uploaded_file')
ext = secure_filename(file.filename).rsplit('.', 1)[1]
if ext in app.config['ALLOWED_EXTENSIONS']:
new_filename = upload_sketch(file, ext,
app.config['S3_BUCKET'],
app.config['UPLOAD_FOLDER'])
return redirect(url_for('uploaded', filename=new_filename))
else:
return render_template("app.html")
def uploaded(filename):
upload_url = s3_url(filename, app.config['S3_BUCKET'],
app.config['UPLOAD_FOLDER'])
model_url = s3_url(filename, app.config['S3_BUCKET'],
app.config['MODEL_FOLDER'])
if request.method == 'GET':
if request.args.get('model_button'):
model_sketch(filename, app.config['S3_BUCKET'],
app.config['UPLOAD_FOLDER'],
app.config['MODEL_FOLDER'])
return render_template("app.html", uploaded_image=upload_url,
model_image=model_url)
else:
return render_template("app.html", uploaded_image=upload_url)
elif request.method == 'POST':
file = request.files.get('uploaded_file')
ext = secure_filename(file.filename).rsplit('.', 1)[1]
if ext in app.config['ALLOWED_EXTENSIONS']:
new_filename = upload_sketch(file, ext,
app.config['S3_BUCKET'],
app.config['UPLOAD_FOLDER'])
return redirect(url_for('uploaded', filename=new_filename))
