Update app.py

app.py

@@ -286,18 +286,26 @@ def get_users():
     return jsonify({"users": [user['username'] for user in users], "count": user_count})
 
 def load_embeddings_from_db():
-    # Collection name 'data' is already correctly used here
-    users = list(mongo.db.data.find())
-    face_data = []# facenet embeddings
-    labels = [] # id 1,2,3,..
-    names = {} #dict of id and roll number
-    # {"1":vinay,"2":shahank}
-    for user in users:
-        face_data.append(user["embeddings"])
-        labels.append(user['id'])  # Keep the ObjectId
-        names[user['id']] = user['RollNumber']  # Use ObjectId as key
-
-    return (face_data, labels, names) if face_data else ([], [], {})
+    if mongo is None:
+        print("MongoDB not connected - returning empty embeddings")
+        return [], [], {}
+    
+    try:
+        users = list(mongo.db.data.find())
+        face_data = []# facenet embeddings
+        labels = [] # id 1,2,3,..
+        names = {} #dict of id and roll number
+        # {"1":vinay,"2":shahank}
+        for user in users:
+            face_data.append(user["embeddings"])
+            labels.append(user['id'])  # Keep the ObjectId
+            names[user['id']] = user['RollNumber']  # Use ObjectId as key
+
+        print(f"Loaded {len(face_data)} user embeddings from database")
+        return (face_data, labels, names) if face_data else ([], [], {})
+    except Exception as e:
+        print(f"Error loading embeddings from database: {e}")
+        return [], [], {}
 
 # Load face embeddings from MongoDB initially
 face_data, labels, names = load_embeddings_from_db()