Update app.py

app.py

@@ -9,10 +9,8 @@ import pickle
 import io
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaFileUpload
-from google_auth_oauthlib.flow import InstalledAppFlow
 from google.auth.transport.requests import Request
 from PIL import Image
-
 from simple_salesforce import Salesforce
 from datetime import datetime
 import uuid
@@ -22,11 +20,10 @@ sf = Salesforce(
     username='suriraja822@agentforce.com',
     password='Sati@1010',
     security_token='B9nS0HEyBE7YmWllqXCyiOJpY',
-    domain='login'  # Use 'test' if you're on a sandbox
+    domain='login'
 )
 
 def log_intruder_to_salesforce(frame_id):
-    """Logs a detection event to Salesforce."""
     try:
         sf.Hotel_Security_Alert__c.create({
             'Alert_Type__c': 'Intruder',
@@ -41,10 +38,6 @@ def log_intruder_to_salesforce(frame_id):
     except Exception as e:
         st.warning(f"⚠️ Failed to log to Salesforce: {e}")
 
-
-
-        
-
 # === CONFIGURATION ===
 API_KEY = "AIzaSyDojJrpauA0XZtCCDUuo9xeQHZQamYKsC4"
 CCTVFEED_IDS = ['1KJRkSf2SKEZ1mXS9_si65IwMBtjs6p4n']
@@ -54,23 +47,18 @@ FRAME_SKIP = st.sidebar.slider("⏩ Frame Skip", 1, 15, 5)
 
 # === SETUP ===
 mp_face_detection = mp.solutions.face_detection
-mp_drawing = mp.solutions.drawing_utils
 face_detector = mp_face_detection.FaceDetection(model_selection=0, min_detection_confidence=0.5)
 
 def mse(img1, img2):
     try:
         if img1.shape != img2.shape:
             img2 = cv2.resize(img2, (img1.shape[1], img1.shape[0]))
-        err = np.mean((img1.astype("float") - img2.astype("float")) ** 2)
-        return err
+        return np.mean((img1.astype("float") - img2.astype("float")) ** 2)
     except Exception:
         return float("inf")
 
 def is_duplicate(img, img_list, threshold=200):
-    for existing in img_list:
-        if mse(img, existing) < threshold:
-            return True
-    return False
+    return any(mse(img, existing) < threshold for existing in img_list)
 
 def get_drive_file_links(folder_id, filter_video=False):
     url = f"https://www.googleapis.com/drive/v3/files?q='{folder_id}'+in+parents&key={API_KEY}&fields=files(id,name,mimeType)"
@@ -145,9 +133,8 @@ def detect_faces_from_video(video_path, registered_faces):
             if not any(mse(f, reg) < 200 for reg in registered_faces) and not is_duplicate(f, unique):
                 unique.append(f)
                 filename = f"intruder_{frame_id}.jpg"
-upload_to_drive(f, filename)
-log_intruder_to_salesforce(frame_id)
-                
+                upload_to_drive(f, filename)
+                log_intruder_to_salesforce(frame_id)
         detected.extend(faces)
     cap.release()
     return detected, unique
@@ -188,4 +175,4 @@ with st.spinner("Processing videos..."):
             os.remove(vfile)
             total_uploaded += len(unique_faces)
 
-st.info(f"🎯 Completed processing all videos.")
+st.info("🎯 Completed processing all videos.")