@@ -61,6 +61,10 @@ def reset_template_state():
6161# Initial page setup
6262#
6363st .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+ )
6468mode = 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 ()
102106except 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