move order hide pipeline ui

src/distilabel_dataset_generator/apps/eval.py

@@ -750,10 +750,6 @@ with gr.Blocks() as app:
     )
 
     btn_push_to_hub.click(
-        fn=hide_pipeline_code_visibility,
-        inputs=[],
-        outputs=[pipeline_code_ui],
-    ).then(
         fn=validate_argilla_user_workspace_dataset,
         inputs=[repo_name],
         outputs=[success_message],
@@ -767,6 +763,10 @@ with gr.Blocks() as app:
         fn=hide_success_message,
         outputs=[success_message],
         show_progress=True,
+    ).success(
+        fn=hide_pipeline_code_visibility,
+        inputs=[],
+        outputs=[pipeline_code_ui],
     ).success(
         fn=push_dataset,
         inputs=[

src/distilabel_dataset_generator/apps/sft.py

@@ -452,7 +452,6 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                         label="Distilabel Pipeline Code",
                     )
 
-
     load_btn.click(
         fn=generate_system_prompt,
         inputs=[dataset_description, temperature],
@@ -465,7 +464,6 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         show_progress=True,
     )
 
-
     btn_apply_to_sample_dataset.click(
         fn=generate_sample_dataset,
         inputs=[system_prompt, num_turns],
@@ -474,10 +472,6 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
     )
 
     btn_push_to_hub.click(
-        fn=hide_pipeline_code_visibility,
-        inputs=[],
-        outputs=[pipeline_code_ui],
-    ).then(
         fn=validate_argilla_user_workspace_dataset,
         inputs=[repo_name],
         outputs=[success_message],
@@ -491,6 +485,10 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         fn=hide_success_message,
         outputs=[success_message],
         show_progress=True,
+    ).success(
+        fn=hide_pipeline_code_visibility,
+        inputs=[],
+        outputs=[pipeline_code_ui],
     ).success(
         fn=push_dataset,
         inputs=[
@@ -517,6 +515,5 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         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

@@ -499,10 +499,6 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
     )
 
     btn_push_to_hub.click(
-        fn=hide_pipeline_code_visibility,
-        inputs=[],
-        outputs=[pipeline_code_ui],
-    ).then(
         fn=validate_argilla_user_workspace_dataset,
         inputs=[repo_name],
         outputs=[success_message],
@@ -516,6 +512,10 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         fn=hide_success_message,
         outputs=[success_message],
         show_progress=True,
+    ).success(
+        fn=hide_pipeline_code_visibility,
+        inputs=[],
+        outputs=[pipeline_code_ui],
     ).success(
         fn=push_dataset,
         inputs=[