improved

main.py

@@ -386,19 +386,25 @@ async def get_random_sample(dataset_name: str = Query(..., alias="dataset-name")
 async def oauth_callback(code: str, state: str):
     # Verify the state value here
 
-    # Exchange the code for a token
-    token_response = requests.post(
-        'https://huggingface.co/oauth/token',
-        data={
-            'grant_type': 'authorization_code',
-            'code': code,
-            'redirect_uri': f'https://{space_host}/login/callback',
-            'client_id': client_id,
-            'client_secret': client_secret
-        }
-    )
-    token_data = token_response.json()
-    access_token = token_data['access_token']
+
+
+    try:
+        token_response = requests.post(
+            'https://huggingface.co/oauth/token',
+            data={
+                'grant_type': 'authorization_code',
+                'code': code,
+                'redirect_uri': f'https://{space_host}/login/callback',
+                'client_id': client_id,
+                'client_secret': client_secret
+            }
+        )
+        token_data = token_response.json()
+        access_token = token_data['access_token']
+
+        print(access_token)
+    except Exception as e:
+        print(e)
 
     # # Fetch user information using access token
     # user_response = requests.get(
@@ -408,9 +414,8 @@ async def oauth_callback(code: str, state: str):
     # user_data = user_response.json()
     # username = user_data['username']
 
-    print(access_token)
 
-    return {"username": username}
+    return {}
 
 @app.get("/oauth-config")
 async def get_oauth_config(request: Request):