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

#define pinDHT 27
#define SCREEN_WIDTH 128 // OLED width in pixels
#define SCREEN_HEIGHT 64 // OLED height in pixels

DHTesp dhtSensor;
Adafruit_SSD1306 oled(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

void setup() {
  Serial.begin(115200);
  dhtSensor.setup(pinDHT, DHTesp::DHT22); // Initialize DHT22 sensor

  if (!oled.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("failed to start SSD1306 OLED"));
    while (true);
  }

  delay(5000); // Delay for 2s for initialization
  oled.clearDisplay(); // Clear display

  oled.setTextSize(1); // Set text size
  oled.setTextColor(WHITE); // Set text color
}

void loop() {
  delay(5000);
  TempAndHumidity data = dhtSensor.getTempAndHumidity();
  float temp = data.temperature;
  float hum = data.humidity;

  if (isnan(hum) || isnan(temp)) {
    Serial.println(F("DHT sensor reading failed!"));
    return;
  }

  oled.clearDisplay(); // Clear display for new data

  // Display temperature and humidity
  oled.setCursor(0, 5);
  oled.print("Temperature: ");
  oled.print(temp);
  oled.print(" C");
  
  oled.setCursor(0, 20);
  oled.print("Humidity: ");
  oled.print(hum);
  oled.print(" %");

  // High Temperature Warning
  if (temp > 30) {
    oled.setCursor(0, 35);
    oled.setTextSize(1);
    oled.print("High Temp Warning!");
  }

  // Low Humidity Warning
  if (hum < 20) {
    oled.setCursor(0, 50);
    oled.setTextSize(1);
    oled.print("Low Humidity Warning!");
  }

  oled.display(); // Display text on OLED

  // Print to Serial Monitor
  Serial.print("Humidity: ");
  Serial.print(hum);
  Serial.print("% \tTemperature: ");
  Serial.print(temp);
  Serial.println(" C");
}