Update src/distilabel_dataset_generator/apps/sft.py

src/distilabel_dataset_generator/apps/sft.py

@@ -1,6 +1,6 @@
 import multiprocessing
 import time
-
+from typings import Union
 import gradio as gr
 import pandas as pd
 from distilabel.distiset import Distiset
@@ -30,6 +30,12 @@ def _run_pipeline(result_queue, num_turns, num_rows, system_prompt):
     distiset: Distiset = pipeline.run(use_cache=False)
     result_queue.put(distiset)
 
+def swap_visibilty(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 generate_system_prompt(dataset_description, progress=gr.Progress()):
     progress(0.1, desc="Initializing text generation")
@@ -298,3 +304,4 @@ with gr.Blocks(
 
     app.load(get_token, outputs=[hf_token])
     app.load(get_org_dropdown, outputs=[org_name])
+    app.load(fn=swap_visibilty, outputs=main_ui)