Vishwanath77 commited on
Commit
063acf6
·
verified ·
1 Parent(s): ef94197

Upload app.py

Browse files
Files changed (1) hide show
  1. src/apps/app.py +4 -4
src/apps/app.py CHANGED
@@ -173,14 +173,14 @@ async def login(response: Response, user: UserLogin, db: AsyncSession = Depends(
173
  access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
174
  access_token = create_access_token(data={"sub": db_user.username, "role": db_user.role}, expires_delta=access_token_expires)
175
 
176
- # Set secure cookie
177
  response.set_cookie(
178
  key="access_token",
179
  value=access_token,
180
  httponly=True,
181
  max_age=ACCESS_TOKEN_EXPIRE_MINUTES * 60,
182
- samesite="lax",
183
- secure=True # Required for Hugging Face HTTPS
184
  )
185
 
186
  print(f"Login successful for {db_user.username}, role: {db_user.role}")
@@ -196,7 +196,7 @@ async def login(response: Response, user: UserLogin, db: AsyncSession = Depends(
196
  async def logout(response: Response):
197
  response.delete_cookie(
198
  key="access_token",
199
- samesite="lax",
200
  secure=True
201
  )
202
  return {"message": "Logged out successfully"}
 
173
  access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
174
  access_token = create_access_token(data={"sub": db_user.username, "role": db_user.role}, expires_delta=access_token_expires)
175
 
176
+ # Set secure cookie - use samesite="none" for Hugging Face iframes
177
  response.set_cookie(
178
  key="access_token",
179
  value=access_token,
180
  httponly=True,
181
  max_age=ACCESS_TOKEN_EXPIRE_MINUTES * 60,
182
+ samesite="none",
183
+ secure=True # Required for Hugging Face HTTPS and samesite="none"
184
  )
185
 
186
  print(f"Login successful for {db_user.username}, role: {db_user.role}")
 
196
  async def logout(response: Response):
197
  response.delete_cookie(
198
  key="access_token",
199
+ samesite="none",
200
  secure=True
201
  )
202
  return {"message": "Logged out successfully"}