Update app.py

app.py

@@ -75,16 +75,26 @@ def update_dataset_with_verified_image(image, encodings_file, label, num_augment
     images_to_encode = [image] + augmented_images
     for img in images_to_encode:
         img_array = np.array(img)
-        encoding = face_recognition.face_encodings(img_array)[0]
-        known_encodings = np.append(known_encodings, [encoding], axis=0)
-        known_labels.append(label)
+        encoding = face_recognition.face_encodings(img_array)
+        if encoding:
+            known_encodings = np.append(known_encodings, [encoding[0]], axis=0)
+            known_labels.append(label)
     save_encodings(known_encodings, known_labels, encodings_file)
 
 def verify_face_with_faiss(image, encodings_file, similarity_threshold=70, num_augmented=5):
     aligned_face = image.convert("RGB")
-    target_encoding = face_recognition.face_encodings(np.array(aligned_face))[0].reshape(1, -1)
+    target_encodings = face_recognition.face_encodings(np.array(aligned_face))
+    
+    if not target_encodings:
+        return False, "No face detected in the provided image. Please try with another image."
+    
+    target_encoding = target_encodings[0].reshape(1, -1)
     
     known_encodings, known_labels = load_encodings(encodings_file)
+    
+    if len(known_encodings) == 0:
+        return False, "No known faces in the database. Please add some faces first."
+    
     known_encodings = np.array(known_encodings)
     
     index = create_faiss_index(known_encodings)
@@ -103,17 +113,26 @@ def verify_face_with_faiss(image, encodings_file, similarity_threshold=70, num_a
     else:
         return False, "No match found."
 
-# Define the Gradio interface
+# Gradio Interface Function
 def gradio_interface(image, similarity_threshold=70):
-    encodings_file = "face_encoding.pkl"
-    result, message = verify_face_with_faiss(image, encodings_file, similarity_threshold=similarity_threshold)
-    return message
+    if image is None:
+        return "Error: No image provided. Please upload an image or take one using the webcam."
+    
+    encodings_file = "./face_encoding.pkl"
+    
+    try:
+        result, message = verify_face_with_faiss(image, encodings_file, similarity_threshold=similarity_threshold)
+        return message
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
-# Launch the Gradio interface
-iface = gr.Interface(fn=gradio_interface, 
-                     inputs=[gr.Image(type="pil"), gr.Slider(0, 100, value=70, label="Similarity Threshold")], 
-                     outputs="text", 
-                     title="Face Recognition with MTCNN and FAISS",
-                     description="Upload an image to see if it matches any face in the database.")
+# Gradio Interface Setup
+iface = gr.Interface(
+    fn=gradio_interface, 
+    inputs=[gr.Image(type="pil"), gr.Slider(0, 100, value=70, label="Similarity Threshold")], 
+    outputs="text", 
+    title="Face Recognition with MTCNN and FAISS",
+    description="Upload an image to see if it matches any face in the database."
+)
 
-iface.launch()
+iface.launch(share=True)