#include <opencv2/opencv.hpp>
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <iostream>

using namespace std;
using namespace cv;

int main(){
    // Load Haar cascades for face and eye detection
    CascadeClassifier face_cascade, eye_cascade;
    if (!face_cascade.load("haarcascade_frontalface_alt.xml") || !eye_cascade.load("haarcascade_eye.xml"))    {
        cout << "Failed to load Haar cascades" << endl;
        return -1;
    }

    // Open camera feed
    VideoCapture cap(0);
    if (!cap.isOpened()){
        cout << "Failed to open camera" << endl;
        return -1;
    }

    // Loop through frames
    while (true){
        Mat frame;
        cap >> frame;

        // Convert frame to grayscale for face and eye detection
        Mat gray_frame;
        cvtColor(frame, gray_frame, COLOR_BGR2GRAY);

        // Detect faces
        vector<Rect> faces;
        face_cascade.detectMultiScale(gray_frame, faces, 1.3, 5);

        // Loop through faces
        for (size_t i = 0; i < faces.size(); i++){
            Rect face_rect = faces[i];

            // Draw rectangle around face
            rectangle(frame, face_rect, Scalar(255, 0, 0), 2);

            // Extract ROI for face
            Mat face_roi = gray_frame(face_rect);

            // Detect eyes within face ROI (region of interest)
            vector<Rect> eyes;
            eye_cascade.detectMultiScale(face_roi, eyes, 1.3, 5);

            // Loop through eyes
            for (size_t j = 0; j < eyes.size(); j++){
                Rect eye_rect = eyes[j];

                // Draw rectangle around eye
                rectangle(frame, Point(face_rect.x + eye_rect.x, face_rect.y + eye_rect.y), 
                          Point(face_rect.x + eye_rect.x + eye_rect.width, face_rect.y + eye_rect.y + eye_rect.height), 
                          Scalar(0, 255, 0), 2);

                // Calculate center of eye
                Point eye_center(face_rect.x + eye_rect.x + eye_rect.width / 2, face_rect.y + eye_rect.y + eye_rect.height / 2);

                // Draw circle at center of eye
                circle(frame, eye_center, 2, Scalar(0, 0, 255), 2);
            }
        }

        // Show frame
        imshow("Gaze Detection", frame);

        // Exit on key press
        if (waitKey(1) == 27) break;
    }

    return 0;
}