Update app.py

app.py

@@ -40,45 +40,35 @@ def get_face(img):
     return None
 
 # Function to verify face (either HOG-SVM or Siamese model)
-def verify(img1, img2, model_type, anchor_img):
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_img1:
-        temp_img1.write(img1.read())
-        temp_img1_path = temp_img1.name
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_img2:
-        temp_img2.write(img2.read())
-        temp_img2_path = temp_img2.name
-
-    img1p = cv2.imread(temp_img1_path)
-    img2p = cv2.imread(temp_img2_path)
-
-    face1 = get_face(img1p)
-    face2 = get_face(img2p)
+def verify(image, model, person):
+    
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_image:
+        temp_image.write(image.read())
+        temp_image_path = temp_image.name
+        
+    image = cv2.imread(temp_image_path)
 
-    if face1 is not None and face2 is not None:
-        st.image([face1, face2], caption=["Image 1", "Image 2"], width=200)
+    face = get_face(image)
 
+    if face is not None:
         if model_type == "HOG-SVM":
-            with open('./svm.pkl', 'rb') as f:
+            with open(f'./svm_{lower(person)}.pkl', 'rb') as f:
                 svm = joblib.load(f)
-            with open('./pca.pkl', 'rb') as f:
+            with open(f'./pca_{lower(person)}.pkl', 'rb') as f:
                 pca = joblib.load(f)
 
-            face1 = preprocess_image_svm(face1)
-            face2 = preprocess_image_svm(face2)
+            face = preprocess_image_svm(face)
 
-            hog1 = extract_hog_features(face1)
-            hog2 = extract_hog_features(face2)
+            hog = extract_hog_features(face)
 
-            hog1_pca = pca.transform([hog1])
-            hog2_pca = pca.transform([hog2])
+            hog_pca = pca.transform([hog])
 
-            pred1 = svm.predict(hog1_pca)
-            pred2 = svm.predict(hog2_pca)
+            pred = svm.predict(hog_pca)
 
-            if pred1 == 1 and pred2 == 1:
-                st.write("Matched")
+            if pred == 1:
+                st.write("Match")
             else:
-                st.write("Not Matched")
+                st.write("Not Match")
     else:
         st.write("Face not detected in one or both images")
 
@@ -86,45 +76,13 @@ def verify(img1, img2, model_type, anchor_img):
 def main():
     st.title("Real-time Face Verification App")
     
-    model_type = st.selectbox("Select Model", ["Siamese", "HOG-SVM"])
-    anchor_img = st.file_uploader("Select Anchor Image", type=["jpg", "png"])
+    model = st.selectbox("Select Model", ["Siamese", "HOG-SVM"])
+    person = st.selectbox("Select Person", ["Theo"])
+    enable = st.checkbox("Enable camera")
+    captured_image = st.camera_input("Take a picture", disabled=not enable)
 
-    if model_type == "Siamese":
-        # Implement Siamese model choice logic here if needed
-        st.write("Using Siamese Network")
-        
-    elif model_type == "HOG-SVM":
-        # Implement HOG-SVM logic here
-        st.write("Using HOG-SVM")
+    if captured_image:
+        verify(captured_image, model, person)
 
-    # Camera Input for Face Detection
-    run_detection = st.checkbox("Start Camera")
-    
-    if run_detection:
-        cap = cv2.VideoCapture(0)  # Start camera
-        
-        while True:
-            ret, frame = cap.read()
-            if not ret:
-                st.write("Failed to grab frame.")
-                break
-            
-            # Detect face in the current frame
-            face = get_face(frame)
-            if face is not None:
-                # Draw bounding box around detected face
-                x1, y1, x2, y2 = face[0], face[1], face[2], face[3]  # Update face coordinates
-                cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
-                
-                # Show bounding box
-                st.image(frame, channels="BGR", use_column_width=True)
-            
-            # Stop camera when ESC is pressed
-            key = cv2.waitKey(1) & 0xFF
-            if key == 27:  # ESC key
-                break
-        
-        cap.release()
-        cv2.destroyAllWindows()
 if __name__ == "__main__":
     main()