Update app.py

app.py

@@ -71,10 +71,28 @@ def setup():
 app, ip_model = setup()
 
 def get_face_landmarks(image_path):
+    def add_padding(image, padding_size=50):
+        height, width = image.shape[:2]
+        padded_image = cv2.copyMakeBorder(
+            image, 
+            top=padding_size, 
+            bottom=padding_size, 
+            left=padding_size, 
+            right=padding_size, 
+            borderType=cv2.BORDER_CONSTANT, 
+            value=[255, 255, 255]  # White padding
+        )
+        return padded_image
+
     detector = dlib.get_frontal_face_detector()
     predictor = dlib.shape_predictor('faceid/shape_predictor_68_face_landmarks.dat')  
+    
     image = cv2.imread(image_path)
     image = imutils.resize(image, width=512)
+
+    # Add padding to the image
+    image = add_padding(image)
+    
     gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     rects = detector(gray, 1)
     for (i, rect) in enumerate(rects):