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

// Definitions
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define DHTPIN 4        // Pin for DHT sensor
#define DHTTYPE DHT22   // DHT sensor type
#define SCREEN_ADDRESS 0x3C // I2C address for OLED

// Create DHT and OLED objects
DHT dht(DHTPIN, DHTTYPE);
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

// Pin LED
const int ledLow = 13;    // Indicator for low temperature
const int ledMedium = 12; // Indicator for medium temperature
const int ledHigh = 11;   // Indicator for high temperature

void setup() {
  Serial.begin(9600);
  dht.begin();

  // Setup LED pins
  pinMode(ledLow, OUTPUT);
  pinMode(ledMedium, OUTPUT);
  pinMode(ledHigh, OUTPUT);

  // Setup OLED display
  display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS);
  display.clearDisplay();
}

void loop() {
  float kel = dht.readHumidity();
  float suh = dht.readTemperature();

  // Check if readings are valid
  if (isnan(kel) || isnan(suh)) {
    Serial.println("Failed to read from DHT sensor!");
    return; // Exit the loop if read fails
  }

  // LED indicators based on temperature
  if (suh < 16) {
    digitalWrite(ledLow, HIGH);
    digitalWrite(ledMedium, LOW);
    digitalWrite(ledHigh, LOW);
  } else if (suh >= 16 && suh <= 27) {
    digitalWrite(ledLow, LOW);
    digitalWrite(ledMedium, HIGH);
    digitalWrite(ledHigh, LOW);
  } else {
    digitalWrite(ledLow, LOW);
    digitalWrite(ledMedium, LOW);
    digitalWrite(ledHigh, HIGH);
  }

  // Display data on OLED
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(WHITE);
  display.setCursor(0, 0);
  display.print("Kelembaban: ");
  display.print(kel);
  display.print(" %");
  display.setCursor(0, 10);
  display.print("Suhu: ");
  display.print(suh);
  display.print(" C");
  display.display();

  // Print to Serial
  Serial.print("Kelembaban: ");
  Serial.print(kel);
  Serial.print(" %; Suhu: ");
  Serial.println(suh);

  delay(2000); // Delay 2 seconds before the next reading
}