#include <LiquidCrystal_I2C.h>
#include <Wire.h>
#include <DHT.h>
#include <ESP32Servo.h>

// Define the type of DHT sensor
#define DHTTYPE DHT22

// Pin definitions
#define DHTPIN 4        // DHT22 data pin connected to GPIO 4
#define OK_PIN 34       // Button OK connected to GPIO 34
#define UP_PIN 35       // Button UP connected to GPIO 35
#define DOWN_PIN 32     // Button DOWN connected to GPIO 32
#define BULB_PIN 15     // Bulb connected to GPIO 15
#define VAP_PIN 16      // Vaporizer connected to GPIO 16
#define SERVO_PIN 17    // Servo connected to GPIO 17

// LCD
LiquidCrystal_I2C lcd(0x27, 16, 2); // Ensure this address is correct for your LCD

// DHT Sensor
DHT dht(DHTPIN, DHTTYPE);

// Servo motor
Servo motor;

// Variables
int ack = 0;
int pos = 0;
int sec = 0;
int Min = 0;
int hrs = 0;
int T_threshold = 30;
int H_threshold = 60;
int SET = 0;
boolean T_condition = true;
boolean H_condition = true;

void setup() {
  lcd.begin(16, 2);
  Serial.begin(9600); // serial data transfer
  lcd.init(); // initialize LCD
  lcd.backlight();

  pinMode(OK_PIN, INPUT_PULLUP);
  pinMode(UP_PIN, INPUT_PULLUP);
  pinMode(DOWN_PIN, INPUT_PULLUP);
  pinMode(BULB_PIN, OUTPUT);
  pinMode(VAP_PIN, OUTPUT);
  
  digitalWrite(BULB_PIN, LOW);
  digitalWrite(VAP_PIN, LOW);

  motor.attach(SERVO_PIN);
  motor.write(pos);

  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Temperature &");
  lcd.setCursor(0, 1);
  lcd.print("Humidity ");
  delay(3000);
  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Controller For");
  lcd.setCursor(0, 1);
  lcd.print("Incubator");
  delay(3000);
  lcd.clear();
  Serial.println("Temperature and Humidity Controller For Incubator");

  dht.begin();
}

void loop() {
  if (SET == 0) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Set Temperature:");
    lcd.setCursor(0, 1);
    lcd.print(T_threshold);
    lcd.print(" *C");
    while (T_condition) {
      if (digitalRead(UP_PIN) == LOW) {
        T_threshold += 1;
        lcd.setCursor(0, 1);
        lcd.print(T_threshold);
        lcd.print(" *C");
        delay(200);
      }
      if (digitalRead(DOWN_PIN) == LOW) {
        T_threshold -= 1;
        lcd.setCursor(0, 1);
        lcd.print(T_threshold);
        lcd.print(" *C");
        delay(200);
      }
      if (digitalRead(OK_PIN) == LOW) {
        delay(200);
        T_condition = false;
      }
    }
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Set Humidity:");
    lcd.setCursor(0, 1);
    lcd.print(H_threshold);
    lcd.print("%");
    delay(100);
    while (H_condition) {
      if (digitalRead(UP_PIN) == LOW) {
        H_threshold += 1;
        lcd.setCursor(0, 1);
        lcd.print(H_threshold);
        lcd.print("%");
        delay(100);
      }
      if (digitalRead(DOWN_PIN) == LOW) {
        H_threshold -= 1;
        lcd.setCursor(0, 1);
        lcd.print(H_threshold);
        lcd.print("%");
        delay(200);
      }
      if (digitalRead(OK_PIN) == LOW) {
        delay(100);
        H_condition = false;
      }
    }
    SET = 1;
  }
  ack = 0;
  
  float temp = dht.readTemperature();
  float hum = dht.readHumidity();
  
  if (isnan(temp) || isnan(hum)) {
    Serial.println("Failed to read from DHT sensor!");
    ack = 1;
  } else {
    ack = 0;
    Serial.print("Temperature: ");
    Serial.print(temp);
    Serial.print(" *C, Humidity: ");
    Serial.print(hum);
    Serial.println(" %");
    
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Temp:");
    lcd.print(temp);
    lcd.print(" *C");
    lcd.setCursor(0, 1);
    lcd.print("Humidity:");
    lcd.print(hum);
    lcd.print(" %");
    
    delay(500);
    if (temp >= T_threshold) {
      digitalWrite(BULB_PIN, LOW);
    } else {
      digitalWrite(BULB_PIN, HIGH);
    }

    if (hum >= H_threshold) {
      digitalWrite(VAP_PIN, HIGH);
    } else {
      digitalWrite(VAP_PIN, LOW);
    }

    sec += 1;
    if (sec == 3) {
      sec = 0;
      Min += 1;
    }
    if (Min == 3) {
      Min = 0;
      hrs += 1;
    }
    if (hrs == 3 && Min == 0 && sec == 0) {
      Serial.println("ROTATING FORWARD");
      for (pos = 0; pos <= 180; pos += 1) {
        motor.write(pos);
        delay(25);
      }
    }
    if (hrs == 6 && Min == 0 && sec == 0) {
      Serial.println("ROTATING BACKWARD");
      hrs = 0;
      for (pos = 180; pos >= 0; pos -= 1) {
        motor.write(pos);
        delay(25);
      }
    }
  }

  if (ack == 1) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("No Sensor data.");
    lcd.setCursor(0, 1);
    lcd.print("System Halted.");
    digitalWrite(BULB_PIN, LOW);
    digitalWrite(VAP_PIN, LOW);
  }
  delay(500);
}