def generate():
if request.method == 'POST':
project_name = request.form.get('project_name')
style = request.form.get('style')
# Prep model on basis of which mdoel is requested
model = prep_model()
# check if the post request has the file part
if 'file' not in request.files:
flash('No file part')
return redirect(request.url)
file = request.files['file']
# if user does not select file, browser still
# submits an empty part without filename
if file.filename == '':
flash('No selected file')
return redirect(request.url)
if file and allowed_file(file.filename):
new_project = Projects(project_name)
db.session.add(new_project)
db.session.commit()
project_id = new_project.id
file_name = str(project_id) + '.png'
file_address = os.path.join(os.getcwd(),
app.config['UPLOAD_FOLDER'], file_name)
file.save(file_address)
output_folder = os.path.join(os.getcwd(),
app.config['OUTPUT_FOLDER'])
html = model.convert_single_image(output_folder,
png_path=file_address,
print_generated_output=0,
get_sentence_bleu=0,
original_gui_filepath=None,
style=style)
project = Projects.get_project_by_id(project_id)
project.html_code = html
project.deploy_url = f'http://localhost:5000/deploy/{project_id}'
db.session.add(project)
db.session.commit()
K.clear_session()
return get_project(project_id)
else:
return render_template('generator_page.html')
def deploy(project_id):
project = Projects.get_project_by_id(project_id)
return project.html_code
def change_html(id):
project = Projects.get_project_by_id(id)
project.html_code = request.form.get('html_code')
db.session.add(project)
db.session.commit()
return get_project(id)
def get_project(id):
return jsonify(Projects.get_project_by_id(id).to_dict())
