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

// OLED Display settings
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET    -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// DHT Sensor settings
#define DHTPIN 15
#define DHTTYPE DHT22
DHT dht(DHTPIN, DHTTYPE);

// Pins
#define POT1 A0
#define POT2 A3
#define BUZZER 27
#define LED_RED 25
#define LED_GREEN 26

void setup() {
  Serial.begin(115200);

  // Initialize OLED
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("OLED initialization failed!"));
    while (true);
  }

  // Initialize pins
  pinMode(LED_RED, OUTPUT);
  pinMode(LED_GREEN, OUTPUT);
  pinMode(BUZZER, OUTPUT);

  dht.begin();

  // Clear OLED on start
  display.clearDisplay();
  display.display();
}

void loop() {
  // Read values
  float temp = dht.readTemperature();
  float hum = dht.readHumidity();
  int pot1 = analogRead(POT1);
  int pot2 = analogRead(POT2);

  // Debug output to Serial Monitor
  Serial.print("Temp: "); Serial.print(temp); Serial.print("°C  ");
  Serial.print("Humidity: "); Serial.print(hum); Serial.print("%  ");
  Serial.print("Pot1: "); Serial.print(pot1); Serial.print("  ");
  Serial.print("Pot2: "); Serial.println(pot2);

  // Display on OLED
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);

  display.print("Temp: "); display.print(temp); display.println(" *C");
  display.print("Hum: "); display.print(hum); display.println(" %");
  display.print("Pot1: "); display.println(pot1);
  display.print("Pot2: "); display.println(pot2);

  // Fault Condition Display
  if (temp > 40 || hum > 80 || pot1 > 3000 || pot2 > 3000) {
    display.println("Status: FAULT");
    digitalWrite(LED_RED, HIGH);
    digitalWrite(LED_GREEN, LOW);
    tone(BUZZER, 1000); // Activate buzzer
  } else {
    display.println("Status: OK");
    digitalWrite(LED_RED, LOW);
    digitalWrite(LED_GREEN, HIGH);
    noTone(BUZZER); // Deactivate buzzer
  }

  display.display(); // Refresh screen
  delay(1000);       // Delay for stability
}