Update app.py

app.py

@@ -1,59 +1,51 @@
-import os
-import cv2
-import numpy as np
-# Define the paths
-staff_folder_path = r"C:\Test1"  # Update to your actual path
-unknown_folder_path = r"C:\Test2"
-
-def compare_images(image1, image2):
-    img1 = cv2.imread(image1)
-    img2 = cv2.imread(image2)
-
-    if img1 is None or img2 is None:
-        return False
-
-    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)")
-
-  # Update to your actual path
-
-if __name__ == "__main__":
-    # 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)
+import os
+import requests
+import json
+from salesforce import Salesforce
+from jwt import encode
+import base64
+from cryptography.hazmat.primitives import serialization
+from cryptography.hazmat.backends import default_backend
+
+# Salesforce credentials
+CONSUMER_KEY = os.environ['SALESFORCE_CONSUMER_KEY']
+CONSUMER_SECRET = os.environ['SALESFORCE_CONSUMER_SECRET']
+USERNAME = os.environ['SALESFORCE_USERNAME']
+PASSWORD = os.environ['SALESFORCE_PASSWORD']
+SECURITY_TOKEN = os.environ['SALESFORCE_SECURITY_TOKEN']
+
+# Custom object and field
+OBJECT_NAME = 'Guest__c'
+FIELD_NAME = 'Adhar_Card_Photo__c'
+
+# Authenticate and get access token
+def get_access_token():
+    payload = {
+        'grant_type': 'password',
+        'client_id': CONSUMER_KEY,
+        'client_secret': CONSUMER_SECRET,
+        'username': USERNAME,
+        'password': PASSWORD + SECURITY_TOKEN
+    }
+    response = requests.post('(link unavailable)', data=payload)
+    return response.json()['access_token']
+
+# Upload image to Salesforce
+def upload_image(image_path, image_name):
+    access_token = get_access_token()
+    sf = Salesforce(instance_url='(link unavailable)', 
+                    session_id=access_token)
+
+    # Read image content
+    with open(image_path, 'rb') as image_file:
+        image_content = image_file.read()
+        image_base64 = base64.b64encode(image_content).decode('utf-8')
+
+    # Create or update record
+    record = sf.query(f"SELECT Id FROM {OBJECT_NAME} WHERE Name = '{image_name}'")
+    if record['records']:
+        record_id = record['records'][0]['Id']
+        sf.update(OBJECT_NAME, {'Id': record_id, FIELD_NAME: image_base64})
+    else:
+        sf.create(OBJECT_NAME, {FIELD_NAME: image_base64, 'Name': image_name})