mominah commited on
Commit
e96f28d
·
verified ·
1 Parent(s): a53a5d2

Update auth.py

Browse files
Files changed (1) hide show
  1. auth.py +4 -1
auth.py CHANGED
@@ -195,10 +195,13 @@ async def logout(request: Request, current_user: dict = Depends(get_current_user
195
  logger.info(f"User logged out: {current_user['email']}")
196
  return {"message": "User logged out successfully"}
197
 
 
 
198
  @router.get("/avatar/{file_id}")
199
  async def get_avatar(file_id: str):
200
  try:
201
- file = fs.get(file_id)
 
202
  return StreamingResponse(file, media_type=file.content_type)
203
  except Exception as e:
204
  logger.error(f"Avatar not found for file_id {file_id}: {e}")
 
195
  logger.info(f"User logged out: {current_user['email']}")
196
  return {"message": "User logged out successfully"}
197
 
198
+ from bson import ObjectId
199
+
200
  @router.get("/avatar/{file_id}")
201
  async def get_avatar(file_id: str):
202
  try:
203
+ # Convert the file_id string to an ObjectId before fetching
204
+ file = fs.get(ObjectId(file_id))
205
  return StreamingResponse(file, media_type=file.content_type)
206
  except Exception as e:
207
  logger.error(f"Avatar not found for file_id {file_id}: {e}")