Update app.py

app.py

@@ -5,34 +5,38 @@ from deepface import DeepFace
 # Path to the dataset folder
 DATASET_PATH = "face_dataset"
 
-# Load dataset images
+# Load dataset images (only image files are included)
 dataset_images = {
-    f: os.path.join(DATASET_PATH, f)
-    for f in os.listdir(DATASET_PATH)
-    if f.lower().endswith(('.jpg', '.jpeg', '.png'))
+    filename: os.path.join(DATASET_PATH, filename)
+    for filename in os.listdir(DATASET_PATH)
+    if filename.lower().endswith(('.jpg', '.jpeg', '.png'))
 }
 
 def recognize_face(uploaded_image):
     """
-    Compare the uploaded image with dataset images and return the best match.
+    Compare the uploaded image with dataset images using DeepFace.
+    Returns the matched model name and its image if a verified match is found.
     """
     if uploaded_image is None:
         return "No image uploaded.", None
 
+    # Loop through each image in the dataset
     for model_name, image_path in dataset_images.items():
         try:
-            # Perform face verification
             result = DeepFace.verify(
-                uploaded_image, image_path, model_name="VGG-Face", enforce_detection=False
+                uploaded_image,
+                image_path,
+                model_name="VGG-Face",
+                enforce_detection=False
             )
-            if result["verified"]:
+            if result.get("verified"):
                 return model_name, image_path
         except Exception as e:
             print(f"Error processing {image_path}: {e}")
 
     return "No matching face found.", None
 
-# Create a Gradio interface
+# Create a Gradio interface for the app
 iface = gr.Interface(
     fn=recognize_face,
     inputs=gr.Image(type="numpy", label="Upload an Image"),