refactor: place Oauth next to push to hub

src/distilabel_dataset_generator/apps/sft.py

@@ -157,7 +157,6 @@ with gr.Blocks(
     title="⚗️ Distilabel Dataset Generator",
     head="⚗️ Distilabel Dataset Generator",
 ) as app:
-    get_login_button()
     gr.Markdown("## Iterate on a sample dataset")
     with gr.Column() as main_ui:
         dataset_description = gr.TextArea(
@@ -210,7 +209,7 @@ with gr.Blocks(
         # Add a header for the full dataset generation section
         gr.Markdown("## Generate full dataset")
         gr.Markdown(
-            "Once you're satisfied with the sample, generate a larger dataset and push it to the hub. Get <a href='https://huggingface.co/settings/tokens' target='_blank'>a Hugging Face token</a> with write access to the organization you want to push the dataset to."
+            "Once you're satisfied with the sample, generate a larger dataset and push it to the hub. Get <a href='https://huggingface.co/settings/tokens' target='_blank'>a Hugging Face token</a> with write access to the organization you want to push the dataset to. A OAuth login resets the session state, so watch out not to lose your generated system prompt!"
         )
 
         with gr.Column() as push_to_hub_ui:
@@ -232,6 +231,7 @@ with gr.Blocks(
                 )
 
             with gr.Row(variant="panel"):
+                get_login_button()
                 hf_token = gr.Textbox(label="HF token", type="password")
                 org_name = get_org_dropdown()
                 repo_name = gr.Textbox(label="Repo name", placeholder="dataset_name")

src/distilabel_dataset_generator/utils.py

@@ -1,5 +1,3 @@
-from typing import Union
-
 import gradio as gr
 from gradio.oauth import (
     OAUTH_CLIENT_ID,
@@ -39,7 +37,7 @@ def get_login_button():
         or get_space() is None
     ):
         return gr.LoginButton(
-            value="Sign in with Hugging Face! (This resets the session)",
+            value="Sign in with Hugging Face! (This resets the session state.)",
             size="lg",
         )
 
@@ -73,20 +71,6 @@ def get_org_dropdown(token: OAuthToken = None):
     )
 
 
-def swap_visibilty(profile: Union[gr.OAuthProfile, None]):
-    if profile is None:
-        return gr.Column(visible=False)
-    else:
-        return gr.Column(visible=True)
-
-
-def swap_visibilty_classes(profile: Union[gr.OAuthProfile, None]):
-    if profile is None:
-        return gr.update(elem_classes=["main_ui_logged_out"])
-    else:
-        return gr.update(elem_classes=["main_ui_logged_in"])
-
-
 def get_token(token: OAuthToken = None):
     if token:
         return token.token