#include <DHT.h>
#include <WiFi.h>
#include <ThingSpeak.h>
#include <Wire.h>
#include <LiquidCrystal_I2C.h>
#include "DHTesp.h"

#define DHT_PIN 2
#define LED_GREEN_PIN 21
#define LED_RED_PIN 22

char ssid[] = "Wokwi-GUEST";
char pass[] = "";
WiFiClient client;

unsigned long myChannelNumber = 2308799;
const char *myWriteAPIKey = "Y5D386LU3W5X66Y2";
int statusCode;
DHTesp dhtSensor;

int ledGreen = LED_GREEN_PIN;
int ledRed = LED_RED_PIN;

struct Data {
  float temperature;
  float humidity;
};

Data data;

// Replace with the address of your LCD
int lcdAddress = 0x27;
int lcdColumns = 16;
int lcdRows = 2;

LiquidCrystal_I2C lcd(lcdAddress, lcdColumns, lcdRows);

void setup() {
  Serial.begin(115200);
  WiFi.mode(WIFI_STA);
  ThingSpeak.begin(client);
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);

  pinMode(ledGreen, OUTPUT);
  pinMode(ledRed, OUTPUT);

  // Initialize the LCD
  lcd.begin(lcdColumns, lcdRows);
  lcd.backlight();  // Turn on the backlight (if available)
  lcd.clear();
}

void connectToCloud() {
  if (WiFi.status() != WL_CONNECTED) {
    Serial.print("Attempting to connect");
    while (WiFi.status() != WL_CONNECTED) {
      WiFi.begin(ssid, pass);
      for (int i = 0; i < 5; i++) {
        Serial.print(".");
        delay(1000);
      }
    }
    Serial.println("\nConnected.");
  }
}

void computeData() {
  TempAndHumidity sensorData = dhtSensor.getTempAndHumidity();
  data.temperature = sensorData.temperature;
  data.humidity = sensorData.humidity;

  // Print data on Serial (optional)
  Serial.println("-----------");
  Serial.println("Humi: " + String(data.humidity));
  Serial.println("Temp: " + String(data.temperature));
  Serial.println("-----------");

  // Display data on LCD
  lcd.setCursor(0, 0);
  lcd.print("Temp: " + String(data.temperature) + "C");

  lcd.setCursor(0, 1);
  lcd.print("Humi: " + String(data.humidity) + "%");
}

void writeData() {
  ThingSpeak.setField(1, data.humidity);
  ThingSpeak.setField(2, data.temperature);
  statusCode = ThingSpeak.writeFields(myChannelNumber, myWriteAPIKey);

  // Display status on LCD
  lcd.setCursor(0, 1);
  lcd.print("Status: " + String(statusCode));

  if (statusCode == 200)
    Serial.println("Channel update successful.");
  else
    Serial.println("Problem Writing data. HTTP error code: " + String(statusCode));
  delay(15000);
}

void loop() {
  connectToCloud();
  computeData();
  writeData();

  float temperature = dhtSensor.getTemperature();
  float humidity = dhtSensor.getHumidity();

  Serial.print("Temperature: ");
  Serial.print(temperature);
  Serial.println("°C");
  Serial.print("Humidity: ");
  Serial.print(humidity);
  Serial.println("%");

  if (temperature > 25.0 && humidity > 70.0) {
    Serial.println("Air quality might be poor (high temperature and humidity).");
    digitalWrite(ledRed, HIGH);
    digitalWrite(ledGreen, LOW);
  } else if (temperature < 20.0 && humidity < 30.0) {
    Serial.println("Air quality might be poor (low temperature and humidity).");
    digitalWrite(ledRed, HIGH);
    digitalWrite(ledGreen, LOW);
  } else {
    Serial.println("Air quality appears to be normal.");
    digitalWrite(ledRed, LOW);
    digitalWrite(ledGreen, HIGH);
  }

  delay(10000);
}