Update app.py

app.py

@@ -1,92 +1,80 @@
 import streamlit as st
 import cv2
-import requests
 import numpy as np
+import requests
 import tempfile
-import time
 import mediapipe as mp
+import os
 
 # === CONFIGURATION ===
-CCTVFEED_IDS = ['1KJRkSf2SKEZ1mXS9_si65IwMBtjs6p4n']  # List of folder IDs (Google Drive)
-SUPPORTED_EXTENSIONS = ['.mp4', '.avi']
+API_KEY = "AIzaSy...your_public_API_key..."  # Replace with a valid public API key if needed
+CCTVFEED_IDS = ['1KJRkSf2SKEZ1mXS9_si65IwMBtjs6p4n']
+REG_FOLDER_ID = '1qkcR7nQTEtiMH9OFUv2bGxVn08E3dKjF'
+INTRUDER_FOLDER_ID = '1PPAUWU-wMx7fek73p-hqPqYQypYtG8Ob'
 
-# === INITIALIZE FACE DETECTION ===
+# === SETUP ===
 mp_face_detection = mp.solutions.face_detection
-face_detection = mp_face_detection.FaceDetection(model_selection=0, min_detection_confidence=0.5)
-
-# === HELPER FUNCTIONS ===
-
-def list_gdrive_folder_files(folder_id):
-    """Returns list of (filename, download_url) from public Google Drive folder."""
-    api_url = f"https://drive.google.com/embeddedfolderview?id={folder_id}#grid"
-    resp = requests.get(api_url)
-    file_lines = resp.text.splitlines()
-    links = []
-    for line in file_lines:
-        if 'data-id=' in line and any(ext in line for ext in SUPPORTED_EXTENSIONS):
-            try:
-                file_id = line.split('data-id="')[1].split('"')[0]
-                file_name = line.split('data-title="')[1].split('"')[0]
-                download_url = f"https://drive.google.com/uc?id={file_id}"
-                links.append((file_name, download_url))
-            except Exception:
-                continue
-    return links
-
-def stream_and_process_video(video_url, filename):
-    """Streams video frame-by-frame and runs face detection."""
-    st.markdown(f"### 🎥 Processing `{filename}`")
-    vid_bytes = requests.get(video_url, stream=True).content
-    tfile = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-    tfile.write(vid_bytes)
-    cap = cv2.VideoCapture(tfile.name)
-
-    frame_count = 0
-    face_count = 0
-
-    frame_display = st.empty()
-    stats_display = st.empty()
+mp_drawing = mp.solutions.drawing_utils
+face_detector = mp_face_detection.FaceDetection(model_selection=0, min_detection_confidence=0.5)
+
+# === FUNCTIONS ===
+def get_drive_video_links(folder_id):
+    api_url = f"https://www.googleapis.com/drive/v3/files?q='{folder_id}'+in+parents&key={API_KEY}&fields=files(id,name,mimeType)"
+    response = requests.get(api_url)
+    files = response.json().get("files", [])
+
+    st.write(f"🔍 Found {len(files)} files in folder `{folder_id}`")
+    for f in files:
+        st.write(f" - **{f['name']}** | `{f['mimeType']}`")
+
+    video_links = []
+    for f in files:
+        if f["mimeType"] in ["video/mp4", "video/x-msvideo", "video/avi"]:
+            link = f"https://drive.google.com/uc?id={f['id']}&export=download"
+            video_links.append((f["name"], link))
+    return video_links
+
+def download_video(link):
+    response = requests.get(link, stream=True)
+    temp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+    for chunk in response.iter_content(chunk_size=1024*1024):
+        temp.write(chunk)
+    temp.close()
+    return temp.name
+
+def detect_faces_from_video(video_path):
+    cap = cv2.VideoCapture(video_path)
+    frames_with_faces = 0
 
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
             break
-        frame_count += 1
-
-        # Convert BGR to RGB
-        image_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        results = face_detection.process(image_rgb)
-
-        # Draw faces
+        rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        results = face_detector.process(rgb)
         if results.detections:
-            for detection in results.detections:
-                bboxC = detection.location_data.relative_bounding_box
-                ih, iw, _ = frame.shape
-                x, y, w, h = int(bboxC.xmin * iw), int(bboxC.ymin * ih), int(bboxC.width * iw), int(bboxC.height * ih)
-                cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
-                face_count += 1
-
-        # Resize for display
-        display_frame = cv2.resize(frame, (640, 360))
-        frame_display.image(display_frame, channels="BGR")
-
-        stats_display.markdown(f"**Frames Processed:** {frame_count} | **Faces Detected So Far:** {face_count}")
-        time.sleep(1 / 24.0)  # Approx. 24 FPS
-
+            frames_with_faces += 1
+            for det in results.detections:
+                mp_drawing.draw_detection(frame, det)
     cap.release()
-    st.success(f"✅ Done! `{filename}`: {frame_count} frames, {face_count} faces detected.")
-
-# === MAIN UI ===
-
-st.title("🚨 CCTV Face Detection from Google Drive")
-st.markdown("Detects faces in `.mp4` and `.avi` videos from your public Google Drive folder.")
-
-for folder_id in CCTVFEED_IDS:
-    videos = list_gdrive_folder_files(folder_id)
-    if not videos:
-        st.warning(f"No valid videos found in folder ID: {folder_id}")
-        continue
-
-    for name, url in videos:
-        if st.button(f"▶️ Run Face Detection on `{name}`"):
-            stream_and_process_video(url, name)
+    return frames_with_faces
+
+# === STREAMLIT UI ===
+st.title("🔐 Intruder Detection from CCTV Feed")
+st.write("Streaming videos from Google Drive, detecting faces using MediaPipe.")
+
+with st.spinner("Fetching videos..."):
+    total_detections = 0
+    for folder_id in CCTVFEED_IDS:
+        videos = get_drive_video_links(folder_id)
+        for vname, link in videos:
+            st.subheader(f"🎥 {vname}")
+            video_file = download_video(link)
+            faces = detect_faces_from_video(video_file)
+            os.remove(video_file)
+            if faces > 0:
+                st.success(f"✅ Detected faces in {faces} frames.")
+                total_detections += 1
+            else:
+                st.warning("⚠️ No faces detected.")
+    st.info(f"🔎 Completed. Videos with detected faces: {total_detections}")