Update app.py

app.py

@@ -1,36 +1,95 @@
-from simple_salesforce import Salesforce, SalesforceMalformedRequest, SalesforceResourceNotFound
-
-# Replace these variables with your Salesforce credentials
-USERNAME = 'alwaysajay411@gmail.com'
-PASSWORD = 'Anusha@310819'
-SECURITY_TOKEN = 'DcroeUKt0Lc4Lpqgzh9dMxEyQ'
-
-# Create a connection to Salesforce
-try:
-    sf = Salesforce(username=USERNAME, password=PASSWORD, security_token=SECURITY_TOKEN)
-
-    # Define your custom object API names (replace with your actual custom object API names)
-    custom_object_1 = 'staff__c'
-    custom_object_2 = 'CustomerData__c'
-
-    # Retrieve records from the first custom object
-    records_custom_object_1 = sf.query(f"SELECT Name, AdhardCard__c, contactNumber__c, pancard__c FROM {custom_object_1}")
-
-    # Retrieve records from the second custom object
-    records_custom_object_2 = sf.query(f"SELECT Name, AdhardCardPhoto__c, pancardPhoto__c, PhoneNumber__c FROM {custom_object_2}")
-
-    # Print retrieved records
-    print("Records from Custom Object 1:")
-    for record in records_custom_object_1['records']:
-        print(record)
-
-    print("\nRecords from Custom Object 2:")
-    for record in records_custom_object_2['records']:
-        print(record)
-
-except SalesforceMalformedRequest as e:
-    print(f"Malformed request: {e}")
-except SalesforceResourceNotFound as e:
-    print(f"Resource not found: {e}")
-except Exception as e:
-    print(f"An error occurred: {e}")  
+import os
+import face_recognition
+import cv2
+import smtplib
+from simple_salesforce import Salesforce
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+
+# Salesforce credentials
+USERNAME = 'your_salesforce_username'
+PASSWORD = 'your_salesforce_password'
+SECURITY_TOKEN = 'your_salesforce_security_token'
+
+# Connect to Salesforce
+sf = Salesforce(username=USERNAME, password=PASSWORD, security_token=SECURITY_TOKEN)
+
+# Query records from two custom objects (with image fields)
+custom_object_1 = 'CustomObject1__c'
+custom_object_2 = 'CustomObject2__c'
+
+# Replace with actual field names
+query_1 = f"SELECT Id, Name, PhotoField__c FROM {custom_object_1}"
+query_2 = f"SELECT Id, Name, PhotoField__c FROM {custom_object_2}"
+
+# Retrieve records
+records_custom_object_1 = sf.query(query_1)['records']
+records_custom_object_2 = sf.query(query_2)['records']
+
+# Combine both objects' records
+salesforce_records = records_custom_object_1 + records_custom_object_2
+
+# Function to send an email if unknown face is found
+def send_email(unknown_image_path):
+    sender_email = "your_email@gmail.com"
+    receiver_email = "receiver_email@gmail.com"
+    password = "your_email_password"
+    
+    message = MIMEMultipart("alternative")
+    message["Subject"] = "Unknown Face Detected"
+    message["From"] = sender_email
+    message["To"] = receiver_email
+    
+    # Email body
+    text = f"An unknown face was detected in the image: {unknown_image_path}"
+    part1 = MIMEText(text, "plain")
+    message.attach(part1)
+    
+    # Send email
+    with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
+        server.login(sender_email, password)
+        server.sendmail(sender_email, receiver_email, message.as_string())
+
+# Folder with input images to match
+input_folder = "path_to_input_images_folder"
+
+# Function to compare images and find matches
+def compare_images(salesforce_records, input_folder):
+    for image_file in os.listdir(input_folder):
+        input_image_path = os.path.join(input_folder, image_file)
+        
+        # Load the input image
+        input_image = face_recognition.load_image_file(input_image_path)
+        input_encoding = face_recognition.face_encodings(input_image)
+
+        if len(input_encoding) == 0:
+            print(f"No face found in {image_file}")
+            continue
+
+        input_encoding = input_encoding[0]
+        
+        found_match = False
+        
+        # Compare input image to Salesforce records
+        for record in salesforce_records:
+            record_photo_url = record['PhotoField__c']
+            
+            # Load Salesforce image (you can download and use images from Salesforce as needed)
+            # Here, assuming the photos are locally available
+            salesforce_image = face_recognition.load_image_file(record_photo_url)
+            salesforce_encoding = face_recognition.face_encodings(salesforce_image)[0]
+            
+            # Compare the encodings
+            matches = face_recognition.compare_faces([salesforce_encoding], input_encoding)
+            
+            if True in matches:
+                print(f"Match found! Input image matches {record['Name']}")
+                found_match = True
+                break
+        
+        if not found_match:
+            print(f"Unknown face in {image_file}")
+            send_email(input_image_path)
+
+# Run the comparison function
+compare_images(salesforce_records, input_folder)