templates.add_template(dataset_key, template)
save_data()
else:
new_template_name = None
dataset_templates = templates.get_templates(dataset_key)
template_list = list(dataset_templates.keys())
if new_template_name:
index = template_list.index(new_template_name)
else:
index = 0
template_name = st.selectbox('', template_list, key='template_select',
index=index, help='Select the template to work on.')
if st.button("Delete Template", key="delete_template"):
templates.remove_template(dataset_key, template.get_name())
save_data("Template deleted!")
#
# If template is selected, displays template editor
#
if template_name is not None:
with st.form("edit_template_form"):
template = dataset_templates[template_name]
code_height = 40
input_fn_code = st.text_area('Input Function', height=code_height, value=template.input_fn)
prompt_fn_code = st.text_area('Prompt Function', height=code_height, value=template.prompt_fn)
output_fn_code = st.text_area('Output Function', height=code_height, value=template.output_fn)
reference = st.text_area("Template Reference",
