#include <DHT.h>
#include <Adafruit_SSD1306.h>
#include <Adafruit_GFX.h>
#include <Fonts/FreeSerif9pt7b.h>
#include <Servo.h>

#define dht_pin 2
#define led_merah 11
#define led_kuning 12
#define led_biru 13
#define buzzer 7
#define servo_pin 8 // Pin for servo
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define DHTTYPE DHT22

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);
DHT dht(dht_pin, DHTTYPE);
Servo myservo;

void setup() {
  Serial.begin(115200);
  dht.begin();
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
  pinMode(buzzer, OUTPUT);
  pinMode(led_merah, OUTPUT);
  pinMode(led_kuning, OUTPUT);
  pinMode(led_biru, OUTPUT);
  myservo.attach(servo_pin); // Attach servo to pin 8
}

void loop() {
  delay(2000);
  display.setFont(&FreeSerif9pt7b);
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(WHITE);
  display.setCursor(0, 40);

  float h = dht.readHumidity();
  float suhu = dht.readTemperature();

  if (isnan(h) || isnan(suhu)) {
    return;
  }

  if (suhu < 29) {
    digitalWrite(led_merah, LOW);
    digitalWrite(led_kuning, LOW);
    digitalWrite(led_biru, HIGH);
    display.println("SUHU RENDAH");
    display.display();
    myservo.write(0); // Set servo to 0 degrees
    noTone(buzzer);
  }
  else if (suhu <= 35 && suhu >= 29) {
    digitalWrite(led_merah, LOW);
    digitalWrite(led_kuning, HIGH);
    digitalWrite(led_biru, LOW);
    display.println("SUHU CUKUP");
    display.display();
    myservo.write(0); // Set servo to 0 degrees
    noTone(buzzer);
  }
  else if (suhu > 35) {
    digitalWrite(led_merah, HIGH);
    digitalWrite(led_kuning, LOW);
    digitalWrite(led_biru, LOW);
    display.println("SUHU PANAS");
    display.display();
    myservo.write(90); // Set servo to 90 degrees
    tone(buzzer, 1000);
    delay(100);
    tone(buzzer, 500);
    delay(100);
  }
  else {
    digitalWrite(led_merah, LOW);
    digitalWrite(led_kuning, LOW);
    digitalWrite(led_biru, LOW);
    display.println("    ");
    display.display();
    myservo.write(0); // Set servo to 0 degrees
    noTone(buzzer);
  }
}