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))