Update app.py

app.py

@@ -27,7 +27,7 @@ def save_hash(enrollment, img_hash):
 
 # Create a consistent hash from image
 def image_hash(image):
-    image = image.resize((100, 100)).convert("L")
+    image = image.resize((100, 100)).convert("L")  # Resize for consistency
     return hashlib.md5(image.tobytes()).hexdigest()
 
 # Main registration logic
@@ -46,17 +46,15 @@ def register_student(image, enrollment, name):
         # Load existing image hashes
         existing_hashes = load_hashes()
 
-        # Check if the same image was registered before (regardless of ID)
-        if new_hash in existing_hashes.values():
-            return "This image is already registered with another Enrollment ID!"
-
-        # Check if enrollment ID already exists with a different image
+        # Check if the same image is already registered with the same ID
         if enrollment in existing_hashes:
             existing_hash = existing_hashes[enrollment]
-            if new_hash != existing_hash:
+            if new_hash == existing_hash:
+                return f"Image already registered with the same image and Enrollment ID {enrollment}."
+            else:
                 return f"Enrollment ID {enrollment} is already used with a different image!"
 
-        # Save image and record hash
+        # If new enrollment, save image and record hash
         image_path = os.path.join(trainimage_path, f"User.{enrollment}.jpg")
         image.save(image_path)
         save_hash(enrollment, new_hash)