Update app.py

app.py

@@ -62,8 +62,9 @@ detector = MTCNN()
 embedder = FaceNet()
 
 # Configure MongoDB
-app.config["MONGO_URI"] = "mongodb+srv://nanduvinay719:76qqKRX4zC97yQun@travis.744fuyn.mongodb.net/?retryWrites=true&w=majority&appName=travis" #config setting
-# Update with your MongoDB URI
+# The URI includes the database name 'travis_db' directly in the path.
+# This is the recommended way to specify the database.
+app.config["MONGO_URI"] = "mongodb+srv://nanduvinay719:76qqKRX4zC97yQun@travis.744fuyn.mongodb.net/travis_db?retryWrites=true&w=majority&appName=travis" 
 mongo = PyMongo(app)#initialize
 haar_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
 
@@ -104,6 +105,7 @@ def cnnlogin():
         return jsonify({"error": "Roll number is required"}), 400
 
     # Fetch user data from MongoDB
+    # Collection name 'data' is already correctly used here
     user_data = mongo.db.data.find_one({"RollNumber": rollnumber}, {"CNN_embeddings": 1, "username": 1})
     if user_data is None:
         return jsonify({"error": "User not found"}), 404
@@ -166,6 +168,7 @@ def recognizeLogin():
     if (results[0]['name']==name):
         print("done")
         today = datetime.now().strftime("%Y-%m-%d")
+        # Collection name 'attendance1' is already correctly used here
         mongo.db.attendance1.update_one(
                 {'username': results[0]['name']},
                 {'$set': {today: True}},  # Mark as present for today
@@ -186,6 +189,7 @@ def register():
 
     print(username)
     # Check if user already exists
+    # Collection name 'data' is already correctly used here
     existing_user = mongo.db.data.find_one({"username": username}, {"_id": 0})
     if existing_user:
         return jsonify({"error": f"User '{username}' already exists"}), 400
@@ -251,6 +255,7 @@ def register():
             print(f"Warning: Could not save model weights in current directory: {e2}")
     
     # Create user data
+    # Collection name 'data' is already correctly used here
     id = mongo.db.data.count_documents({}) + 1
     user_data = {
         'RollNumber': rollnumber,
@@ -265,6 +270,7 @@ def register():
 
     # Insert into MongoDB
     mongo.db.data.insert_one(user_data)
+    # Collection name 'attendance1' is already correctly used here
     mongo.db.attendance1.insert_one({"username": rollnumber, "id": id})
     # Reload embeddings
     reload_embeddings()
@@ -274,11 +280,13 @@ def register():
 # Load embeddings from MongoDB for recognition
 @app.route('/get_users', methods=['GET'])
 def get_users():
+    # Collection name 'data' is already correctly used here
     users = list(mongo.db.data.find({}, {"id": 1, "username": 1}))  # Fetch only _id and username
     user_count = len(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,..
@@ -385,6 +393,7 @@ def recognize_faces_in_image(image):
 @app.route('/users/<username>/images', methods=['GET'])
 def get_user_images(username):
     username = str(username).upper()
+    # Collection name 'data' is already correctly used here
     details = mongo.db.data.find_one({"username": username},{"_id":0,"embeddings":0,"CNN_embeddings":0})
     if not details:
         details = mongo.db.data.find_one({"RollNumber": username},{"_id":0,"embeddings":0,"CNN_embeddings":0})
@@ -411,6 +420,7 @@ def recognize():
     today = datetime.now().strftime("%Y-%m-%d")
     for result in results:
         if result['name'] != "Unknown":  # Only log attendance for recognized users
+            # Collection name 'attendance1' is already correctly used here
             mongo.db.attendance1.update_one(
                 {'username': result['name']},
                 {'$set': {today: True}},  # Mark as present for today
@@ -438,11 +448,13 @@ def user_recognize():
 @app.route('/user_attendance/<username>', methods=['GET'])
 def get_user_attendance(username):
     # Check if the user exists in the database
+    # Collection name 'data' is already correctly used here
     user = mongo.db.data.find_one({'RollNumber': username})
     if user is None:
         return jsonify({"error": "User not found"}), 404
     print(user['username'])
     # Fetch the attendance data for the user
+    # Collection name 'attendance1' is already correctly used here
     attendance = mongo.db.attendance1.find_one({'username': username}, {'_id': 0,"username":0,"id":0})
     print(attendance)
     if attendance is None:
@@ -454,8 +466,6 @@ def get_user_attendance(username):
 #attendance
 @app.route('/attendance',methods=['GET'])
 def get_attendance():
+    # Collection name 'attendance1' is already correctly used here
     records = list(mongo.db.attendance1.find({}, {"_id": 0}))
-    return jsonify({"attendance": records})
-
-# if __name__ == '__main__':
-#     app.run(debug=True)
+    return jsonify({"attendance": records})