Update app.py

app.py

@@ -324,6 +324,95 @@ class IntrusionDetection:
         except Exception as e:
             raise ValueError(f"Error in detect_intrusion: {str(e)}")
 
+class IntrusionDetectionEn:
+    def __init__(self, model_path="yolov8n.pt", max_intrusion_time=300, iou_threshold=0.5, conf_threshold=0.7):
+        self.model_path = model_path
+        self.max_intrusion_time = max_intrusion_time
+        self.iou_threshold = iou_threshold
+        self.conf_threshold = conf_threshold
+
+        # Predefined staff uniform colors (RGB format)
+        self.staff_colors = [
+            (139, 143, 133),  # Grayish tone
+            (146, 150, 140),  # Light grayish tone
+            (146, 152, 141),  # Muted gray-green
+            (143, 147, 136),  # Gray-green
+            (48, 59, 71)      # Dark blue/gray
+        ]
+
+    def is_staff(self, person_crop):
+        """Checks if the detected person is a staff member based on clothing color."""
+        avg_color = np.mean(person_crop, axis=(0, 1))  # Compute average color (BGR)
+        avg_color = avg_color[::-1]  # Convert BGR to RGB
+
+        # Compute Euclidean distance to known staff colors
+        for color in self.staff_colors:
+            dist = np.linalg.norm(np.array(avg_color) - np.array(color))
+            if dist < 30:  # Threshold to consider it a match
+                return True
+        return False
+
+    def intrusion_detect_en(self, video_path):
+        try:
+            device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+            if not os.path.exists(self.model_path):
+                model = YOLO("yolov8n.pt")
+                model.save(self.model_path)
+            else:
+                model = YOLO(self.model_path)
+            model.to(device)
+
+            cap = cv2.VideoCapture(video_path)
+            if not cap.isOpened():
+                raise ValueError(f"❌ Failed to open video: {video_path}")
+
+            fps = int(cap.get(cv2.CAP_PROP_FPS))
+            width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+            height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+            output_path = "output_intrusion.mp4"
+            fourcc = cv2.VideoWriter_fourcc(*"mp4v")
+            out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+            if not out.isOpened():
+                cap.release()
+                raise ValueError(f"❌ Failed to initialize video writer")
+
+            frame_count = 0
+
+            while cap.isOpened():
+                ret, frame = cap.read()
+                if not ret:
+                    break
+                frame_count += 1
+
+                results = model(frame)
+                for result in results:
+                    boxes = result.boxes.xyxy.cpu().numpy()
+                    classes = result.boxes.cls.cpu().numpy()
+                    confidences = result.boxes.conf.cpu().numpy()
+
+                    for box, cls, conf in zip(boxes, classes, confidences):
+                        if int(cls) == 0 and conf > self.conf_threshold:  # Person class
+                            x1, y1, x2, y2 = map(int, box)
+                            person_crop = frame[y1:y2, x1:x2]
+
+                            if self.is_staff(person_crop):
+                                continue  # Ignore staff members
+
+                            label = "Intruder"
+                            color = (0, 0, 255)
+                            cv2.rectangle(frame, (x1, y1), (x2, y2), color, 2)
+                            cv2.putText(frame, label, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
+
+                out.write(frame)
+
+            cap.release()
+            out.release()
+            if frame_count == 0 or not os.path.exists(output_path):
+                raise ValueError("❌ Processing failed: No frames processed or output not created")
+            return output_path
+        except Exception as e:
+            raise ValueError(f"Error in detect_intrusion: {str(e)}")
 
 import cv2
 import numpy as np
@@ -481,6 +570,7 @@ def process_video(feature, video, area=None):
         "Fall Detection": FallDetection,
         "Fight Detection": FightDetection,
         "Intrusion Detection": IntrusionDetection,
+        "Intrusion Detection En" : IntrusionDetectionEn, 
         "Loitering Detection": LoiteringDetection,
         "Fire And Smoke Detection": FireAndSmokeDetection
     }
@@ -504,7 +594,7 @@ interface = gr.Interface(
     inputs=[
         gr.Dropdown(choices=[
             "Crowd Detection", "Fall Detection", 
-            "Fight Detection", "Intrusion Detection", "Loitering Detection",
+            "Fight Detection", "Intrusion Detection", "Intrusion Detection En", "Loitering Detection",
             "Fire And Smoke Detection"
         ], label="Select Feature"),
         gr.Video(label="Upload Video"),