Update app.py

app.py

@@ -9,7 +9,8 @@ from googleapiclient.http import MediaIoBaseDownload, MediaFileUpload
 from PIL import Image
 import torch
 import io
-from deepface import DeepFace  # Use DeepFace for advanced face recognition
+from deepface import DeepFace
+from scipy.spatial.distance import cosine  # Import for cosine similarity calculation
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
@@ -42,7 +43,7 @@ cphotos_folder_id = '1DGeRqRbCPcfLDdEgP0h5fyX-MF8EQ8AH'
 suspects_folder_id = '1N3RMhVD0OygeufLPYod6IYLtqzvlm3Jv'
 
 # Limit the number of threads for scalability
-MAX_THREADS = 5
+MAX_THREADS = 3
 
 def list_files_in_folder(folder_id):
     logging.debug(f"Listing files in folder ID: {folder_id}")
@@ -89,9 +90,10 @@ def extract_face_embedding(image_path):
         return None
 
 def calculate_similarity(encoding1, encoding2):
-    # Use cosine similarity for comparison
-    similarity = DeepFace.find(img_path1=encoding1, img_path2=encoding2, distance_metric='cosine')[0]
-    return 1 - similarity  # Convert distance to similarity
+    # Use cosine similarity directly
+    distance = cosine(encoding1, encoding2)
+    similarity = 1 - distance
+    return similarity
 
 def compare_faces_with_encodings(encoding_cctv, aadhar_encodings, threshold=0.7):
     for aadhar_name, aadhar_encoding in aadhar_encodings: