update successful message and pipeline code

src/distilabel_dataset_generator/apps/base.py

@@ -300,25 +300,6 @@ def get_iterate_on_sample_dataset_ui(
     )
 
 
-def get_pipeline_code_ui(pipeline_code: str) -> gr.Code:
-    gr.Markdown("## Customize and run with distilabel")
-    gr.HTML("<hr>")
-
-    with gr.Accordion(
-        "Run this pipeline using distilabel",
-        open=False,
-    ):
-        gr.Markdown(
-            "You can run this pipeline locally with distilabel. For more information, please refer to the [distilabel documentation](https://distilabel.argilla.io/) or go to the FAQ tab at the top of the page for more information."
-        )
-        pipeline_code = gr.Code(
-            value=pipeline_code,
-            language="python",
-            label="Distilabel Pipeline Code",
-        )
-    return pipeline_code
-
-
 def get_argilla_tab() -> Tuple[Any]:
     with gr.Tab(label="Argilla"):
         if get_argilla_client() is not None:
@@ -492,7 +473,7 @@ def get_success_message_row() -> gr.Markdown:
     return success_message
 
 
-def show_success_message_hub(org_name, repo_name) -> gr.Markdown:
+def show_success_message(org_name, repo_name) -> gr.Markdown:
     client = get_argilla_client()
     argilla_api_url = client.api_url
     return gr.Markdown(
@@ -500,25 +481,27 @@ def show_success_message_hub(org_name, repo_name) -> gr.Markdown:
         <div style="padding: 1em; background-color: #e6f3e6; border-radius: 5px; margin-top: 1em;">
             <h3 style="color: #2e7d32; margin: 0;">Dataset Published Successfully!</h3>
             <p style="margin-top: 0.5em;">
-                Your dataset is now available the Hugging Face Hub:
-                <a href="https://huggingface.co/datasets/{org_name}/{repo_name}" target="_blank" style="color: #1565c0; text-decoration: none;">
-                    https://huggingface.co/datasets/{org_name}/{repo_name}
-                </a>
+                <strong>
+                    <a href="{argilla_api_url}" target="_blank" style="color: #1565c0; text-decoration: none;">
+                        Open dataset in Argilla space
+                    </a>
+                </strong>
             </p>
             <p style="margin-top: 0.5em;">
-                Your dataset is now available within Argilla:
-                <a href="{argilla_api_url}" target="_blank" style="color: #1565c0; text-decoration: none;">
-                    {argilla_api_url}
+                The generated dataset is in the right format for fine-tuning with TRL, AutoTrain, or other frameworks. Your dataset is now available at: 
+                <a href="https://huggingface.co/datasets/{org_name}/{repo_name}" target="_blank" style="color: #1565c0; text-decoration: none;">
+                    https://huggingface.co/datasets/{org_name}/{repo_name}
                 </a>
-                <br>Unfamiliar with Argilla? Here are some docs to help you get started:
-                <br>• <a href="https://docs.argilla.io/latest/how_to_guides/annotate/" target="_blank">How to curate data in Argilla</a>
-                <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>
             </p>
         </div>
+        <p style="margin-top: 1em; font-size: 0.9em; color: #333;">
+            Unfamiliar with Argilla? Here are some docs to help you get started:
+            <br>• <a href="https://docs.argilla.io/latest/how_to_guides/annotate/" target="_blank">How to curate data in Argilla</a>
+            <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>
+        </p>
         """,
         visible=True,
     )
 
-
 def hide_success_message() -> gr.Markdown:
     return gr.Markdown(value="")

src/distilabel_dataset_generator/apps/sft.py

@@ -10,9 +10,8 @@ from distilabel.distiset import Distiset
 from huggingface_hub import HfApi
 
 from src.distilabel_dataset_generator.apps.base import (
-    get_pipeline_code_ui,
     hide_success_message,
-    show_success_message_hub,
+    show_success_message,
     validate_argilla_user_workspace_dataset,
     validate_push_to_hub,
 )
@@ -343,6 +342,10 @@ def push_dataset_to_argilla(
     return ""
 
 
+def update_pipeline_code_visibility():
+    return {pipeline_code_ui: gr.Accordion(visible=True)}
+
+
 ######################
 # Gradio UI
 ######################
@@ -394,7 +397,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
 
         gr.HTML(value="<hr>")
         gr.Markdown(value="## 3. Generate your dataset")
-        with gr.Row():
+        with gr.Row(equal_height=False):
             with gr.Column(scale=2):
                 org_name = get_org_dropdown()
                 repo_name = gr.Textbox(
@@ -417,11 +420,22 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                 )
                 btn_push_to_hub = gr.Button("Push to Hub", variant="primary", scale=2)
             with gr.Column(scale=3):
-                success_message = gr.Markdown()
-
-        pipeline_code = get_pipeline_code_ui(
-            generate_pipeline_code(system_prompt.value, num_turns.value, num_rows.value)
-        )
+                success_message = gr.Markdown(visible=True)
+                with gr.Accordion(
+                    "Do you want to go further? Customize and run with Distilabel",
+                    open=False,
+                    visible=False,
+                ) as pipeline_code_ui:
+                    code = generate_pipeline_code(
+                        system_prompt=system_prompt.value,
+                        num_turns=num_turns.value,
+                        num_rows=num_rows.value,
+                    )
+                    pipeline_code = gr.Code(
+                        value=code,
+                        language="python",
+                        label="Distilabel Pipeline Code",
+                    )
 
     gr.on(
         triggers=[load_btn.click, btn_apply_to_sample_dataset.click],
@@ -463,9 +477,14 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         outputs=[success_message],
         show_progress=True,
     ).success(
-        fn=show_success_message_hub,
+        fn=show_success_message,
         inputs=[org_name, repo_name],
         outputs=[success_message],
+    ).success(
+        fn=update_pipeline_code_visibility,
+        inputs=[],
+        outputs=[pipeline_code_ui],
     )
+
     app.load(fn=swap_visibility, outputs=main_ui)
     app.load(fn=get_org_dropdown, outputs=[org_name])

src/distilabel_dataset_generator/apps/textcat.py

@@ -10,9 +10,8 @@ from distilabel.distiset import Distiset
 from huggingface_hub import HfApi
 
 from src.distilabel_dataset_generator.apps.base import (
-    get_pipeline_code_ui,
     hide_success_message,
-    show_success_message_hub,
+    show_success_message,
     validate_argilla_user_workspace_dataset,
     validate_push_to_hub,
 )
@@ -351,6 +350,9 @@ def validate_input_labels(labels):
 def update_max_num_labels(labels):
     return gr.update(maximum=len(labels) if labels else 1)
 
+def update_pipeline_code_visibility():
+    return {pipeline_code_ui: gr.Accordion(visible=True)}
+
 
 ######################
 # Gradio UI
@@ -438,7 +440,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
 
         gr.HTML("<hr>")
         gr.Markdown("## 3. Generate your dataset")
-        with gr.Row():
+        with gr.Row(equal_height=False):
             with gr.Column(scale=2):
                 org_name = get_org_dropdown()
                 repo_name = gr.Textbox(
@@ -462,17 +464,24 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                 btn_push_to_hub = gr.Button("Push to Hub", variant="primary", scale=2)
             with gr.Column(scale=3):
                 success_message = gr.Markdown(visible=True)
-
-        pipeline_code = get_pipeline_code_ui(
-            generate_pipeline_code(
-                system_prompt.value,
-                difficulty=difficulty.value,
-                clarity=clarity.value,
-                labels=labels.value,
-                num_labels=num_labels.value,
-                num_rows=num_rows.value,
-            )
-        )
+                with gr.Accordion(
+                    "Do you want to go further? Customize and run with Distilabel",
+                    open=False,
+                    visible=False,
+                ) as pipeline_code_ui:
+                    code = generate_pipeline_code(
+                        system_prompt.value,
+                        difficulty=difficulty.value,
+                        clarity=clarity.value,
+                        labels=labels.value,
+                        num_labels=num_labels.value,
+                        num_rows=num_rows.value,
+                    )
+                    pipeline_code = gr.Code(
+                        value=code,
+                        language="python",
+                        label="Distilabel Pipeline Code",
+                    )
 
     gr.on(
         triggers=[load_btn.click, btn_apply_to_sample_dataset.click],
@@ -525,9 +534,14 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         outputs=[success_message],
         show_progress=True,
     ).success(
-        fn=show_success_message_hub,
+        fn=show_success_message,
         inputs=[org_name, repo_name],
         outputs=[success_message],
+    ).success(
+        fn=update_pipeline_code_visibility,
+        inputs=[],
+        outputs=[pipeline_code_ui],
     )
+
     app.load(fn=swap_visibility, outputs=main_ui)
     app.load(fn=get_org_dropdown, outputs=[org_name])