get username

main.py

@@ -392,7 +392,8 @@ async def oauth_callback(code: str, state: str):
     credentials_bytes = credentials.encode("ascii")
     base64_credentials = base64.b64encode(credentials_bytes)
     auth_header = f"Basic {base64_credentials.decode('ascii')}"
-
+    username = ""
+    
     try:
         token_response = requests.post(
             'https://huggingface.co/oauth/token',
@@ -409,6 +410,30 @@ async def oauth_callback(code: str, state: str):
         if token_response.status_code == 200:
             tokens = token_response.json()
             access_token = tokens.get('access_token')
+
+
+            if access_token:
+                    try:
+                        user_response = requests.get(
+                            'https://huggingface.co/api/user',
+                            headers={'Authorization': f'Bearer {access_token}'}
+                        )
+
+                        if user_response.status_code == 200:
+                            user_data = user_response.json()
+                            username = user_data['username']
+                            # Now you have the username, you can handle it as needed
+                        else:
+                            username = ""
+                            print(f"Error getting user data: {user_response.status_code}, {user_response.text}")
+
+                    except Exception:
+                        traceback.print_exc()
+                        username = ""
+            else:
+                username = ""
+
+
             # ID Token can be extracted here if needed
             # id_token = tokens.get('id_token')
         else:
@@ -418,7 +443,7 @@ async def oauth_callback(code: str, state: str):
         traceback.print_exc()
         access_token = ""
 
-    return {"access_token": access_token}
+    return {"access_token": access_token, "username" : username}
 
 @app.get("/oauth-config")
 async def get_oauth_config(request: Request):