#include <WiFi.h>
#include <Adafruit_SSD1306.h> // OLED display library

// === Pin Definitions ===
const int redLedPin = 15;     // System active LED (Red)
const int greenLedPin = 2;    // Face recognized LED (Green)
const int buzzerPin = 12;     // Buzzer for unrecognized faces
const int sensorPwrPin = 23;  // Sensor power pin

// === Display Initialization ===
Adafruit_SSD1306 display(128, 64, &Wire, -1);

// === Timing Variables ===
unsigned long previousMillis = 0;
const long interval = 5000; // 5 seconds interval

// === Simulated Web Command ===
bool htmlStartDetectionPressed = true; // Simulates the HTML "Start Detection" button

void setup()
{
  Serial.begin(115200);

  // Initialize pins
  pinMode(sensorPwrPin, OUTPUT);
  pinMode(buzzerPin, OUTPUT);
  pinMode(greenLedPin, OUTPUT);
  pinMode(redLedPin, OUTPUT);

  // Turn everything off initially
  digitalWrite(buzzerPin, LOW);
  digitalWrite(greenLedPin, LOW);
  digitalWrite(redLedPin, LOW);

  // Initialize OLED
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    while (true); // Halt
  }

  // Show splash screen
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 10);
  display.println("Face");
  display.setCursor(0, 35);
  display.println("Recognition");
  display.display();
  delay(2000);

  display.clearDisplay();
  display.setTextSize(1);
  display.setCursor(0, 0);
  display.println("AI Robot Booting...");
  display.display();
  delay(1000);

  Serial.println("Face Recognition AI Robot Started.");
  Serial.println("Awaiting command from HTML UI...");

  // System ready indicator
  digitalWrite(redLedPin, HIGH);
}

void loop()
{
  unsigned long currentMillis = millis();

  if (currentMillis - previousMillis >= interval)
  {
    previousMillis = currentMillis;

    if (htmlStartDetectionPressed) // Simulate user clicking "Start Detection"
    {
      Serial.println("[HTML] Command received: Start Detection");

      bool faceRecognized = recognizeFace(); // Simulated face recognition

      if (faceRecognized)
      {
        digitalWrite(greenLedPin, HIGH);
        digitalWrite(buzzerPin, LOW);
        displayAttendanceStatus("Access Granted");
        Serial.println("[AI Robot] Face recognized - Access Granted.");
        Serial.println("[HTML] Sending status to UI: Face recognized ✔️");
      }
      else
      {
        digitalWrite(greenLedPin, LOW);
        digitalWrite(buzzerPin, HIGH);
        displayAttendanceStatus("Access Denied");
        Serial.println("[AI Robot] Face not recognized - Access Denied.");
        Serial.println("[HTML] Sending status to UI: Face not recognized ❌");
      }
    }
    else
    {
      Serial.println("[HTML] Waiting for user to start detection...");
    }
  }
}

// === Simulate Face Recognition ===
bool recognizeFace()
{
  return random(0, 2) == 1; // Random true or false
}

// === Display Result on OLED ===
void displayAttendanceStatus(String status)
{
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 20);
  display.println(status);
  display.display();
}