Spaces:
Running
Running
Update auth.py
Browse files
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 |
-
|
|
|
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}")
|