Added cognito authentication - Fix 3

app/main.py

@@ -17,19 +17,18 @@ async def protected_route(user = Depends(get_current_user)):
 @app.get("/auth/callback")
 async def auth_callback(code: str):
     try:
-        # Exchange the authorization code for tokens
         tokens = exchange_code_for_token(code)
         
-        # Create a response with the access token
+        # Use id_token instead of access_token
         response = JSONResponse(content={
             "message": "Authentication successful",
-            "access_token": tokens["access_token"]
+            "id_token": tokens["id_token"]  # Changed from access_token
         })
         
-        # Set the access token as a cookie
+        # Store id_token in cookie
         response.set_cookie(
-            key="access_token",
-            value=tokens["access_token"],
+            key="token",
+            value=tokens["id_token"],  # Changed from access_token
             httponly=True,
             secure=True,
             samesite="lax"