Update server.py

server.py

@@ -211,19 +211,17 @@ async def login(request: Request):
 
 @app.route("/auth")
 async def auth(request: Request):
-    try:
-        token = await oauth.google.authorize_access_token(request)
-        print("TOKEN:", token)
-        id_token = token.get("id_token")
-        if id_token is not None:
-            user_info = await oauth.google.parse_id_token(request, token)
-        else:
+    token = await oauth.google.authorize_access_token(request)
+    print("TOKEN:", token)
+    id_token = token.get("id_token")
+    if id_token:
+        user_info = await oauth.google.parse_id_token(request, token)
+    else:
+        user_info = token.get("userinfo")
+        if not user_info:
             user_info = await oauth.google.userinfo(request, token=token)
-        request.session["user"] = dict(user_info)
-        return RedirectResponse(url="/")
-    except Exception as e:
-        print("OAuth error:", e)
-        return PlainTextResponse(f"OAuth error: {e}")
+    request.session["user"] = dict(user_info)
+    return RedirectResponse(url="/")
 
 @app.route("/logout")
 async def logout(request: Request):