Update app.py

app.py

@@ -2,17 +2,54 @@ import os
 import cv2
 import numpy as np
 
-staff_folder_path = r"C:\Users\nikit\OneDrive\Desktop\Test1"  # Replace with the correct path
-unknown_folder_path = r"C:\Users\nikit\OneDrive\Desktop\Test2"  # Replace with the correct path
+def compare_images(image1, image2):
+    img1 = cv2.imread(image1)
+    img2 = cv2.imread(image2)
 
-print(f"Staff folder path: {staff_folder_path}")
-print(f"Unknown folder path: {unknown_folder_path}")
-import os
+    if img1 is None or img2 is None:
+        return False
 
-if os.path.exists(path):
-    print("Path exists!")
-else:
-    print("Path does not exist.")
+    img1 = cv2.resize(img1, (100, 100))
+    img2 = cv2.resize(img2, (100, 100))
+
+    gray1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
+    gray2 = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
+
+    score, _ = cv2.compare_ssim(gray1, gray2, full=True)
     
+    return score >= 0.9
+
+def compare_folders(staff_folder, unknown_folder):
+    if not os.path.exists(staff_folder):
+        print(f"Error: Staff folder '{staff_folder}' does not exist.")
+        return
 
+    if not os.path.exists(unknown_folder):
+        print(f"Error: Unknown folder '{unknown_folder}' does not exist.")
+        return
+
+    for staff_image in os.listdir(staff_folder):
+        staff_image_path = os.path.join(staff_folder, staff_image)
+        found_match = False
+        
+        for unknown_image in os.listdir(unknown_folder):
+            unknown_image_path = os.path.join(unknown_folder, unknown_image)
+            
+            if compare_images(staff_image_path, unknown_image_path):
+                print(f"Match found: {staff_image} matches {unknown_image} (Staff)")
+                found_match = True
+                break
+        
+        if not found_match:
+            print(f"No match found for {staff_image} (Unknown)")
+
+# Define the paths
+staff_folder_path = r"C:\Users\nikit\OneDrive\Desktop\Test1"  # Update to your actual path
+unknown_folder_path = r"C:\Users\nikit\OneDrive\Desktop\Test2"  # Update to your actual path
+
+# Print paths for debugging
+print(f"Staff folder path: {staff_folder_path}")
+print(f"Unknown folder path: {unknown_folder_path}")
 
+# Compare folders
+compare_folders(staff_folder_path, unknown_folder_path)