Skip to content

Commit d239a8a

Browse files
authored
align vocabulary in app with the paper (#550)
* align vocabulary in app with the paper * strip input and target templates * prompt/template a few missing * repo link less proeminent
1 parent 00b0f0b commit d239a8a

File tree

1 file changed

+42
-39
lines changed

1 file changed

+42
-39
lines changed

promptsource/app.py

Lines changed: 42 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,10 @@ def reset_template_state():
6161
# Initial page setup
6262
#
6363
st.set_page_config(page_title="Promptsource", layout="wide")
64+
st.sidebar.markdown(
65+
"<center><a href='https://github.com/bigscience-workshop/promptsource'>💻Github - Promptsource\n\n</a></center>",
66+
unsafe_allow_html=True,
67+
)
6468
mode = st.sidebar.selectbox(
6569
label="Choose a mode",
6670
options=select_options,
@@ -101,7 +105,7 @@ def show_text(t, width=WIDTH, with_markdown=False):
101105
template_collection = TemplateCollection()
102106
except FileNotFoundError:
103107
st.error(
104-
"Unable to find the templates folder!\n\n"
108+
"Unable to find the prompt folder!\n\n"
105109
"We expect the folder to be in the working directory. "
106110
"You might need to restart the app in the root directory of the repo."
107111
)
@@ -166,33 +170,32 @@ def get_infos(d_name):
166170
results.append(
167171
{
168172
"Dataset name": dataset_name,
169-
"Subset name": "" if subset_name is None else subset_name,
173+
"Subset name": "" if subset_name is None else subset_name,
170174
"Train size": split_sizes["train"] if "train" in split_sizes else 0,
171175
"Validation size": split_sizes["validation"] if "validation" in split_sizes else 0,
172176
"Test size": split_sizes["test"] if "test" in split_sizes else 0,
173-
"Number of templates": len(dataset_templates),
174-
"Number of original task templates": sum(
177+
"Number of prompts": len(dataset_templates),
178+
"Number of original task prompts": sum(
175179
[bool(t.metadata.original_task) for t in dataset_templates.templates.values()]
176180
),
177-
"Template names": [t.name for t in dataset_templates.templates.values()],
178-
# TODO: template name is not very informative... refine that
181+
"Prompt names": [t.name for t in dataset_templates.templates.values()],
179182
}
180183
)
181184
results_df = pd.DataFrame(results)
182-
results_df.sort_values(["Number of templates"], inplace=True, ascending=False)
185+
results_df.sort_values(["Number of prompts"], inplace=True, ascending=False)
183186
results_df.reset_index(drop=True, inplace=True)
184187

185188
nb_training_instances = results_df["Train size"].sum()
186189
st.write(f"## Number of *training instances*: `{nb_training_instances}`")
187190

188-
plot_df = results_df[["Dataset name", "Subset name", "Train size", "Number of templates"]].copy()
191+
plot_df = results_df[["Dataset name", "Subset name", "Train size", "Number of prompts"]].copy()
189192
plot_df["Name"] = plot_df["Dataset name"] + " - " + plot_df["Subset name"]
190193
plot_df.sort_values(["Train size"], inplace=True, ascending=False)
191194
fig = px.bar(
192195
plot_df,
193196
x="Name",
194197
y="Train size",
195-
hover_data=["Dataset name", "Subset name", "Number of templates"],
198+
hover_data=["Dataset name", "Subset name", "Number of prompts"],
196199
log_y=True,
197200
title="Number of training instances per data(sub)set - y-axis is in logscale",
198201
)
@@ -269,19 +272,19 @@ def get_infos(d_name):
269272
template_list = dataset_templates.all_template_names
270273
num_templates = len(template_list)
271274
st.sidebar.write(
272-
"No of Templates created for "
275+
"No of prompts created for "
273276
+ f"`{dataset_key + (('/' + conf_option.name) if conf_option else '')}`"
274277
+ f": **{str(num_templates)}**"
275278
)
276279

277280
if mode == "Prompted dataset viewer":
278281
if num_templates > 0:
279282
template_name = st.sidebar.selectbox(
280-
"Template name",
283+
"Prompt name",
281284
template_list,
282285
key="template_select",
283286
index=0,
284-
help="Select the template to visualize.",
287+
help="Select the prompt to visualize.",
285288
)
286289

287290
step = 50
@@ -336,14 +339,14 @@ def get_infos(d_name):
336339
#
337340
if num_templates > 0:
338341
template = dataset_templates[template_name]
339-
st.subheader("Template")
342+
st.subheader("Prompt")
340343
st.markdown("##### Name")
341344
st.text(template.name)
342345
st.markdown("##### Reference")
343346
st.text(template.reference)
344347
st.markdown("##### Original Task? ")
345348
st.text(template.metadata.original_task)
346-
st.markdown("##### Choices in prompt? ")
349+
st.markdown("##### Choices in template? ")
347350
st.text(template.metadata.choices_in_prompt)
348351
st.markdown("##### Metrics")
349352
st.text(", ".join(template.metadata.metrics) if template.metadata.metrics else None)
@@ -352,13 +355,13 @@ def get_infos(d_name):
352355
show_jinja(template.get_answer_choices_expr())
353356
else:
354357
st.text(None)
355-
st.markdown("##### Jinja")
358+
st.markdown("##### Jinja template")
356359
splitted_template = template.jinja.split("|||")
357-
st.markdown("###### Prompt + X")
358-
show_jinja(splitted_template[0])
360+
st.markdown("###### Input template")
361+
show_jinja(splitted_template[0].strip())
359362
if len(splitted_template) > 1:
360-
st.markdown("###### Y")
361-
show_jinja(splitted_template[1])
363+
st.markdown("###### Target template")
364+
show_jinja(splitted_template[1].strip())
362365
st.markdown("***")
363366

364367
#
@@ -378,14 +381,14 @@ def get_infos(d_name):
378381
if prompt == [""]:
379382
st.write("∅∅∅ *Blank result*")
380383
else:
381-
st.write("Prompt + X")
384+
st.write("Input")
382385
show_text(prompt[0])
383386
if len(prompt) > 1:
384-
st.write("Y")
387+
st.write("Target")
385388
show_text(prompt[1])
386389
st.markdown("***")
387390
else: # mode = Sourcing
388-
st.markdown("## Template Creator")
391+
st.markdown("## Prompt Creator")
389392

390393
#
391394
# Create a new template or select an existing one
@@ -402,20 +405,20 @@ def get_infos(d_name):
402405

403406
with col1a, st.form("new_template_form"):
404407
new_template_name = st.text_input(
405-
"Create a New Template",
408+
"Create a New Prompt",
406409
key="new_template",
407410
value="",
408-
help="Enter name and hit enter to create a new template.",
411+
help="Enter name and hit enter to create a new prompt.",
409412
)
410413
new_template_submitted = st.form_submit_button("Create")
411414
if new_template_submitted:
412415
if new_template_name in dataset_templates.all_template_names:
413416
st.error(
414-
f"A template with the name {new_template_name} already exists "
417+
f"A prompt with the name {new_template_name} already exists "
415418
f"for dataset {state.templates_key}."
416419
)
417420
elif new_template_name == "":
418-
st.error("Need to provide a template name.")
421+
st.error("Need to provide a prompt name.")
419422
else:
420423
template = Template(new_template_name, "", "")
421424
dataset_templates.add_template(template)
@@ -424,18 +427,18 @@ def get_infos(d_name):
424427
else:
425428
state.new_template_name = None
426429

427-
with col1b, st.beta_expander("or Select Template", expanded=True):
430+
with col1b, st.beta_expander("or Select Prompt", expanded=True):
428431
dataset_templates = template_collection.get_dataset(*state.templates_key)
429432
template_list = dataset_templates.all_template_names
430433
if state.template_name:
431434
index = template_list.index(state.template_name)
432435
else:
433436
index = 0
434437
state.template_name = st.selectbox(
435-
"", template_list, key="template_select", index=index, help="Select the template to work on."
438+
"", template_list, key="template_select", index=index, help="Select the prompt to work on."
436439
)
437440

438-
if st.button("Delete Template", key="delete_template"):
441+
if st.button("Delete Prompt", key="delete_prompt"):
439442
dataset_templates.remove_template(state.template_name)
440443
reset_template_state()
441444

@@ -466,8 +469,8 @@ def get_infos(d_name):
466469
with st.form("edit_template_form"):
467470
updated_template_name = st.text_input("Name", value=template.name)
468471
state.reference = st.text_input(
469-
"Template Reference",
470-
help="Short description of the template and/or paper reference for the template.",
472+
"Prompt Reference",
473+
help="Short description of the prompt and/or paper reference for the prompt.",
471474
value=template.reference,
472475
)
473476

@@ -476,12 +479,12 @@ def get_infos(d_name):
476479
state.metadata.original_task = st.checkbox(
477480
"Original Task?",
478481
value=template.metadata.original_task,
479-
help="Template asks model to perform the original task designed for this dataset.",
482+
help="Prompt asks model to perform the original task designed for this dataset.",
480483
)
481484
state.metadata.choices_in_prompt = st.checkbox(
482-
"Choices in Prompt?",
485+
"Choices in Template?",
483486
value=template.metadata.choices_in_prompt,
484-
help="Template explicitly lists choices in the prompt for the output.",
487+
help="Prompt explicitly lists choices in the template for the output.",
485488
)
486489

487490
# Metrics from here:
@@ -510,7 +513,7 @@ def get_infos(d_name):
510513
metrics_choices,
511514
default=template.metadata.metrics,
512515
help="Select all metrics that are commonly used (or should "
513-
"be used if a new task) to evaluate this template.",
516+
"be used if a new task) to evaluate this prompt.",
514517
)
515518

516519
# Answer choices
@@ -535,11 +538,11 @@ def get_infos(d_name):
535538
and updated_template_name != state.template_name
536539
):
537540
st.error(
538-
f"A template with the name {updated_template_name} already exists "
541+
f"A prompt with the name {updated_template_name} already exists "
539542
f"for dataset {state.templates_key}."
540543
)
541544
elif updated_template_name == "":
542-
st.error("Need to provide a template name.")
545+
st.error("Need to provide a prompt name.")
543546
else:
544547
# Parses state.answer_choices
545548
if state.answer_choices == "":
@@ -569,10 +572,10 @@ def get_infos(d_name):
569572
if prompt == [""]:
570573
st.write("∅∅∅ *Blank result*")
571574
else:
572-
st.write("Prompt + X")
575+
st.write("Input")
573576
show_text(prompt[0], width=40)
574577
if len(prompt) > 1:
575-
st.write("Y")
578+
st.write("Target")
576579
show_text(prompt[1], width=40)
577580

578581

0 commit comments

Comments
 (0)