davidberenstein1957 HF staff commited on
Commit
5c6644f
·
1 Parent(s): 88a4065

update success message based on argilla availability

Browse files
src/distilabel_dataset_generator/apps/base.py CHANGED
@@ -475,6 +475,27 @@ def get_success_message_row() -> gr.Markdown:
475
 
476
  def show_success_message(org_name, repo_name) -> gr.Markdown:
477
  client = get_argilla_client()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
478
  argilla_api_url = client.api_url
479
  return gr.Markdown(
480
  value=f"""
 
475
 
476
  def show_success_message(org_name, repo_name) -> gr.Markdown:
477
  client = get_argilla_client()
478
+ if client is None:
479
+ return gr.Markdown(
480
+ value="""
481
+ <div style="padding: 1em; background-color: #e6f3e6; border-radius: 5px; margin-top: 1em;">
482
+ <h3 style="color: #2e7d32; margin: 0;">Dataset Published Successfully!</h3>
483
+ <p style="margin-top: 0.5em;">
484
+ The generated dataset is in the right format for fine-tuning with TRL, AutoTrain, or other frameworks. Your dataset is now available at:
485
+ <a href="https://huggingface.co/datasets/{org_name}/{repo_name}" target="_blank" style="color: #1565c0; text-decoration: none;">
486
+ https://huggingface.co/datasets/{org_name}/{repo_name}
487
+ </a>
488
+ </p>
489
+ <p style="margin-top: 1em; font-size: 0.9em; color: #333;">
490
+ By configuring an `ARGILLA_API_URL` and `ARGILLA_API_KEY` you can curate the dataset in Argilla.
491
+ Unfamiliar with Argilla? Here are some docs to help you get started:
492
+ <br>• <a href="https://docs.argilla.io/latest/getting_started/quickstart/" target="_blank">How to get started with Argilla</a>
493
+ <br>• <a href="https://docs.argilla.io/latest/how_to_guides/annotate/" target="_blank">How to curate data in Argilla</a>
494
+ <br>• <a href="https://docs.argilla.io/latest/how_to_guides/import_export/" target="_blank">How to export data once you have reviewed the dataset</a>
495
+ </p>
496
+ </div>
497
+ """
498
+ )
499
  argilla_api_url = client.api_url
500
  return gr.Markdown(
501
  value=f"""