fixing issues in face-try.py file

Author: HappySaxena

face-try.py

@@ -5,23 +5,53 @@
 @author: Lenovo
 """
 import cv2
+import sys
 
-face_cascade = cv2.CascadeClassifier('E:\\project\\main\\models\\haarcascade_frontalface_default.xml')
+# Load Haar cascade
+cascade_path = '/home/happy/gssoc/driver-drowsiness-detection-system/models/haarcascade_frontalface_default.xml'
+face_cascade = cv2.CascadeClassifier(cascade_path)
 
+# Check if cascade loaded successfully
+if face_cascade.empty():
+    print(f"[ERROR] Failed to load cascade from {cascade_path}")
+    sys.exit(1)
+
+# Start video capture
 cap = cv2.VideoCapture(0)
 
-while cap.isOpened():
-    _, img = cap.read()
+if not cap.isOpened():
+    print("[ERROR] Cannot access webcam.")
+    sys.exit(1)
+
+try:
+    while True:
+        ret, img = cap.read()
+
+        # Check if frame was captured
+        if not ret or img is None:
+            print("[WARNING] Failed to read frame from webcam.")
+            continue
+
+        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+
+        try:
+            faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=4)
+        except cv2.error as e:
+            print(f"[ERROR] detectMultiScale failed: {e}")
+            continue
+
+        for (x, y, w, h) in faces:
+            cv2.rectangle(img, (x, y), (x + w, y + h), (255, 0, 0), 3)
 
-    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    faces = face_cascade.detectMultiScale(gray, 1.1, 4)
+        cv2.imshow('Face Detection', img)
 
-    for (x, y , w ,h) in faces:
-        cv2.rectangle(img, (x,y), (x+w, y+h), (255, 0 , 0), 3)
+        if cv2.waitKey(1) & 0xFF == ord('q'):
+            print("[INFO] Exiting on user request.")
+            break
 
-    cv2.imshow('Face Detection', img)
-    if cv2.waitKey(1) & 0xFF == ord('q'):
-        break
+except KeyboardInterrupt:
+    print("\n[INFO] Exiting on Ctrl+C")
 
-cap.release()
-cv2.destroyAllWindows()
+finally:
+    cap.release()
+    cv2.destroyAllWindows()