checked and verified endpoints

.gitignore

@@ -8,6 +8,7 @@
 *.so
 *.dylib
 
+**/__pycache__/
 # Ignore build output directories
 /bin/
 /build/

src/api/endpoints.py

@@ -2,7 +2,6 @@
 from werkzeug.utils import secure_filename
 from face_recognition_model import FaceRecognizer
 from preprocessing import Preprocessor
-from search_embeddings import Search_Embeddings
 import mysql.connector
 import cv2
 import numpy as np  
@@ -40,8 +39,10 @@ def recognize_face():
 
     preprocessor = Preprocessor(None, 160, 160, 3)
 
-    recognizer = FaceRecognizer('src/models/weights/facenet.tflite')
-
+    script_dir = os.path.dirname(__file__) # get the directory of the current script
+    src_dir= os.path.dirname(script_dir)
+    model_path = os.path.join(script_dir, 'models/weights/facenet.tflite')
+    recognizer = FaceRecognizer(model_path)
     img = preprocessor.face_detect(img) # Detect face
     img = preprocessor.face_align(img) # Align Face
     img = preprocessor.crop_face_image(img) # Crop face image
@@ -99,7 +100,11 @@ def register_user():
     image_array = np.array(image)
     # Process the image and get the embedding
     preprocessor = Preprocessor(None, 160, 160, 3)
-    recognizer = FaceRecognizer('src/models/weights/facenet.tflite') 
+    script_dir = os.path.dirname(__file__) # get the directory of the current script
+    src_dir= os.path.dirname(script_dir)
+    model_path = os.path.join(script_dir, 'models/weights/facenet.tflite')
+    recognizer = FaceRecognizer(model_path)
+
 
     img = preprocessor.face_detect(image_array)  # Detect face
     img = preprocessor.face_align(img)  # Align face

src/api/preprocessing.py

@@ -45,7 +45,7 @@ def face_align(self, image):
                 dx = right_eye[0] - left_eye[0]
                 angle = np.degrees(np.arctan2(dy, dx))
                 # Rotate the image to align the eyes horizontally
-                M = cv2.getRotationMatrix2D(((left_eye[0] + right_eye[0]) // 2, (left_eye[1] + right_eye[1]) // 2), angle, 1)
+                M = cv2.getRotationMatrix2D((int((left_eye[0] + right_eye[0]) // 2), int((left_eye[1] + right_eye[1]) // 2)), angle, 1)
                 image = cv2.warpAffine(image, M, (self.width, self.height))
         return image