Made changes for face login

users/routes.py

@@ -8,6 +8,7 @@ from users.services import create_user_account, create_user_embeddings, update_u
 from services.facial_processing import FacialProcessing
 from services.face_match import FaceMatch
 import os
+import tempfile
 from datetime import timedelta
 from dotenv import load_dotenv
 from auth.services import get_token
@@ -25,76 +26,93 @@ async def create_user(data: UserCreate, db: Session = Depends(get_db)):
     new_user = await create_user_account(data, db)
     return new_user
 
-@router.get("/me", response_model=UserBase)
+@router.get("/me/", response_model=UserBase)
 async def read_users_me(current_user: User = Depends(get_current_user)):
     return current_user
 
-@router.put("/me", response_model=UserBase)
+@router.put("/me/", response_model=UserBase)
 async def update_user_me(user: UserUpdate, current_user: User = Depends(get_current_user), db: Session = Depends(get_db)):
     updated_user = update_user(db, current_user.id, user)
     return updated_user
 
-@router.post("/me/face", status_code=status.HTTP_200_OK)
+@router.post("/me/face/", status_code=status.HTTP_200_OK)
 async def create_face_embeddings(file: UploadFile = File(...), user: User = Depends(get_current_user), db: Session = Depends(get_db)):
     face_processor = FacialProcessing()
 
-    image_path = f"faces/{user.id}.jpg"
-    with open(image_path, "wb") as buffer:
-        buffer.write(await file.read())
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        temp_file.write(await file.read())
+        temp_file.flush()
+        image_path = temp_file.name 
+
 
     embeddings = face_processor.extract_embeddings_vgg(image_path)
     if embeddings:
         user_embeddings = UserEmbeddingsBase(embeddings=embeddings)
         await create_user_embeddings(user.id, user_embeddings, db)
+        
+        os.remove(image_path)
+
         return {"message": "Face embeddings created successfully"}
 
+    os.remove(image_path)
+
     raise HTTPException(status_code=400, detail="Failed to process face")
 
-@router.get("/me/face", status_code=status.HTTP_200_OK)
+@router.get("/me/face/", status_code=status.HTTP_200_OK)
 async def get_face_embeddings(user: User = Depends(get_current_user), db: Session = Depends(get_db)):
     face = db.query(UserEmbeddingsBase).filter(UserEmbeddingsBase.user_id == user.id).first()
     if not face:
         raise HTTPException(status_code=404, detail="Face embeddings not found")
     return JSONResponse(content={"embeddings": face.embeddings}, status_code=status.HTTP_200_OK)
 
-@router.put("/me/face", status_code=status.HTTP_200_OK)
+@router.put("/me/face/", status_code=status.HTTP_200_OK)
 async def update_face_embeddings(file: UploadFile = File(...), user: User = Depends(get_current_user), db: Session = Depends(get_db)):
     face_processor = FacialProcessing()
 
-    image_path = f"faces/{user.id}.jpg"
-    with open(image_path, "wb") as buffer:
-        buffer.write(await file.read())
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        temp_file.write(await file.read())
+        temp_file.flush()
+        image_path = temp_file.name
 
     embeddings = face_processor.extract_embeddings_vgg(image_path)
     if embeddings:
         user_embeddings = UserEmbeddingsBase(embeddings=embeddings)
         await update_user_embeddings(user.id, user_embeddings, db)
+        
+        os.remove(image_path)
         return {"message": "Face embeddings updated successfully"}
 
+    os.remove(image_path)
     raise HTTPException(status_code=400, detail="Failed to process face")
 
-@router.post("/login/face")
+@router.post("/login/face/")
 async def face_login(file: UploadFile = File(...), db: Session = Depends(get_db)):
     face_processor = FacialProcessing()
     face_matcher = FaceMatch(db)
 
-    image_path = f"temp_{file.filename}"
-    with open(image_path, "wb") as buffer:
-        buffer.write(await file.read())
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        temp_file.write(await file.read())
+        temp_file.flush()
+        image_path = temp_file.name
 
     embeddings = await face_processor.extract_embeddings(image_path)
     if not embeddings:
+        os.remove(image_path)
         raise HTTPException(status_code=400, detail="Failed to process face")
     
     match_result = face_matcher.new_face_matching(embeddings)
     if match_result['status'] == 'Success':
         user = get_user_by_id(match_result['user_id'], db)
         if not user:
+            os.remove(image_path)
             raise HTTPException(status_code=404, detail="User not found")
         
         access_token_expires = timedelta(minutes=int(os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES", "30")))
         payload = {"id": user.id, "sub": user.email}
         token = get_token(payload, db)
+
+        os.remove(image_path)
         return JSONResponse(content=token.dict(), status_code=status.HTTP_200_OK)
     
+    os.remove(image_path)
     raise HTTPException(status_code=401, detail="Face not recognized")