Update app.py

app.py

@@ -7,7 +7,7 @@ from dataset_uploader import ParquetScheduler
 # Setup  #
 ##########
 
-contributor_username = whoami()["name"]
+
 
 # only show an info the first time uploading to the hub
 show_info = True 
@@ -105,11 +105,11 @@ def setup_submission(system_prompt="", history=[], chat_format="sharegpt"):
     return history
 
 
-def save_sft_data(system_prompt="", history=[], sft_chat_format="sharegpt"):
+def save_sft_data(system_prompt="", history=[], sft_chat_format="sharegpt", oauth_token: gr.OAuthToken | None):
     """
     A function that pushes the data to the hub.
     """
-
+    contributor_username = whoami(oauth_token.token)["name"]
     # setup the info message to only show once
     global show_info
     scheduler = schedulers[f"sft-{sft_chat_format}"]
@@ -142,12 +142,12 @@ def save_sft_data(system_prompt="", history=[], sft_chat_format="sharegpt"):
 
 
 def save_dpo_data(
-    system_prompt="", history=[], chosen="", rejected="", dpo_chat_format="sharegpt"
+    system_prompt="", history=[], chosen="", rejected="", dpo_chat_format="sharegpt", oauth_token: gr.OAuthToken | None
 ):
     """
     A function that pushes the data to the hub.
     """
-
+    contributor_username = whoami(oauth_token.token)["name"]
     # setup the info message to only show once
     global show_info
     scheduler = schedulers[f"dpo-{dpo_chat_format}"]
@@ -201,7 +201,7 @@ def undo_chat(history):
 # Interface  #
 ##############
 
-with gr.Blocks() as demo:
+with gr.Blocks() as interface:
     gr.Markdown("<h1 style='text-align: center'>ShareGPT-Builder</h1>")
 
     #### SFT ####
@@ -287,5 +287,19 @@ with gr.Blocks() as demo:
             height="560px"
             ></iframe>""")
 
+
+def swap_visibilty(profile: gr.OAuthProfile | None):
+    return gr.update(elem_classes=["main_ui_logged_in"]) if profile else gr.update(elem_classes=["main_ui_logged_out"])
+        
+css = '''
+.main_ui_logged_out{opacity: 0.3; pointer-events: none}
+'''
+
+with gr.Blocks(css=css) as demo:
+    gr.LoginButton()
+    with gr.Column(elem_classes="main_ui_logged_out") as main_ui:
+        interface.render()
+    demo.load(fn=swap_visibilty, outputs=main_ui)
+
 if __name__ == "__main__":
     demo.launch(debug=True, show_error=True)