#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <MPU6050_light.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

MPU6050 mpu(Wire);

#define BUTTON_PIN 4
#define BUZZER_PIN 12

void setup() {
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  pinMode(BUZZER_PIN, OUTPUT);

  Wire.begin();
  Serial.begin(115200);

  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("OLED not found"));
    while (1);
  }
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);

  mpu.begin();
  mpu.calcOffsets();
  delay(1000);
}

void loop() {
  mpu.update();

  bool eyesClosed = digitalRead(BUTTON_PIN) == LOW;
  bool headTilted = abs(mpu.getAngleX()) > 30 || abs(mpu.getAngleY()) > 30;

  display.clearDisplay();
  display.setCursor(0, 0);
  display.println("Drowsiness Monitor");

  if (eyesClosed && headTilted) {
    display.println("ALERT: Drowsy!");
    digitalWrite(BUZZER_PIN, HIGH);
  } else {
    display.println("Status: Normal");
    digitalWrite(BUZZER_PIN, LOW);
  }

  display.display();
  delay(500);
}