Update app.py

app.py

@@ -1,11 +1,37 @@
 import gradio as gr
 import spaces
-#import sys
-#import os
 import training
+from huggingface_hub import HfApi
+from huggingface_hub import whoami
 
 model = training.model_training()
 
+# Get top-level authorizations
+oauth_info = {'username' : None, 'token' : None}
+api = None
+
+def get_oauth_info(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
+    global oauth_info
+    global api
+
+    print(f'profile = {profile}')
+    if profile is None:
+        oauth_info['username'] = None
+        oauth_info['token'] = None
+        api = HfApi()
+        
+        return "Please login to the Huggingface with login button"
+    else:
+        #print(f'Testing {profile} for username')
+        oauth_info['username'] = profile.username
+        oauth_info['token'] = oauth_token.token
+
+        api = HfApi(token=oauth_token.token)
+        org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
+
+        return print(f'{profile.username}: {org_names}')
+
+
 def get_dataset():
     global dataset_cfg
 
@@ -18,8 +44,9 @@ def run_training():
     return model.run_training()
 
 with gr.Blocks() as demo:
+    gr.LoginButton()
     text1 = gr.Markdown("Starting to test SyNet")
-    text2 = gr.Markdown("")
+    user_text = gr.Markdown("")
 
     load_btn = gr.Button("Load dataset")
     load_text = gr.Markdown("")
@@ -30,6 +57,7 @@ with gr.Blocks() as demo:
     load_btn.click(get_dataset, inputs=None, outputs=[load_text])
     train_btn.click(run_training, inputs=None, outputs=[train_text])
 
+    demo.load(get_oauth_info, inputs=None, outputs=user_text)
 
 if __name__ == "__main__":
     demo.launch()