Add vector_search function for pipeline aggregation

API/database.py

@@ -22,3 +22,29 @@ def find_one_and_delete(self, collection, query):
 
     def update_one(self, collection, query, update):
         return self.db[collection].update_one(query, update)
+
+    # add a function for pipeline aggregation vector search
+    def vector_search(self, collection, embedding):
+
+        result = self.db[collection].aggregate([
+            {
+                "$vectorSearch": {
+                "index": "vector_index",
+                "path": "face_embedding",
+                "queryVector": embedding,
+                "numCandidates": 20,
+                "limit": 20
+                }
+            }, {
+                '$project': {
+                '_id': 0, 
+                'Name': 1,
+                'Image': 1,
+                'score': {
+                    '$meta': 'vectorSearchScore'
+                    }
+                }
+            }
+            ])
+        result_arr = [i for i in result]
+        return result_arr

API/route.py

@@ -8,22 +8,27 @@
 
 from bson import ObjectId
 from deepface import DeepFace
-from fastapi import APIRouter, HTTPException, Response
+from fastapi import APIRouter, HTTPException, Response, UploadFile, File
 from matplotlib import pyplot as plt
 from PIL import Image
 from pydantic import BaseModel
 
 from API.database import Database
 from API.utils import init_logging_config
+from dotenv import load_dotenv
 
+load_dotenv()
 init_logging_config()
 
+MONGO_URI = os.getenv("MONGO_URL1")
 router = APIRouter()
 
 
 client = Database()
+client2 = Database(MONGO_URI, "FaceRec")
 
 collection = "faceEntries"
+collection2 = "ImageDB"
 
 
 # Models  for the data to be sent and received by the server
@@ -267,3 +272,37 @@ async def delete_employees(EmployeeCode: int):
     client.find_one_and_delete(collection, {"EmployeeCode": EmployeeCode})
 
     return {"Message": "Successfully Deleted"}
+
+
+@router.post("/recognize_face", response_class=Response)
+async def recognize_face(Face: UploadFile = File(...)):
+    """
+    Recognize a face from the provided image.
+
+    Args:
+        Face (UploadFile): The image file to be recognized.
+
+    Returns:
+        Response: A response object containing the recognized employee information in JSON format.
+
+    Raises:
+        HTTPException: If an internal server error occurs.
+    """
+    logging.info("Recognizing Face")
+    try:
+        img_data = await Face.read()
+        with open("temp.png", "wb") as f:
+            f.write(img_data)
+
+        embedding = DeepFace.represent(img_path="temp.png", model_name="Facenet")
+        result = client2.vector_search(collection2, embedding[0]['embedding'])
+        logging.info(f"Result: {result}")
+        os.remove("temp.png")
+    except Exception as e:
+        logging.error(f"Error: {e}")
+        os.remove("temp.png")
+        raise HTTPException(status_code=500, detail="Internal server error")
+    return Response(
+        content=bytes(json.dumps(result[0], default=str), "utf-8"),
+        media_type="application/json",
+    )

CHANGELOG.md

@@ -49,4 +49,17 @@
 - Resolved various bugs and issues identified during the testing process.
 
 ### Removed
-- Removed deprecated code and unused dependencies from the project.
+- Removed deprecated code and unused dependencies from the project.
+
+## [0.1.4] - 2024-03-16 - 23:00
+
+### Added
+- Implemented a new `recognize_Face` endpoint in [`route.py`](API/route.py). This endpoint accepts a base64 string as input, converts it into embeddings, and performs a vector search query on the MongoDB Atlas database. Changes made by @Devasy23.
+- Added a new `vector_search` function in [`database.py`](API/database.py). This function performs a vector similarity search on the MongoDB Atlas database using Euclidean distance as the similarity measure. Changes made by @Devasy23.
+- Updated [`index.ipynb`](index.ipynb) to include examples and usage of the new `recognize_Face` endpoint and `vector_search` function. Changes made by @Devasy23.
+
+### Changed
+- Updated the `Database` class in [`database.py`](API/database.py) to include the new `vector_search` function. Changes made by @Devasy23.
+
+### Fixed
+- Resolved various bugs and issues identified during the implementation and testing of the new features. Fixes made by @Devasy23.