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

#define SCREEN_WIDTH 128  // OLED display width
#define SCREEN_HEIGHT 64  // OLED display height
#define OLED_RESET -1     // Reset pin (not used)

// Initialize OLED display
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

#define TEMP_SENSOR_PIN 34   // Simulating Temperature Sensor
#define SMOKE_SENSOR_PIN 35  // Simulating Smoke Sensor
#define RED_LED 27           // Fire Alert LED
#define GREEN_LED 26         // Normal LED
#define BUZZER 25            // Buzzer for Alarm

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

    pinMode(RED_LED, OUTPUT);
    pinMode(GREEN_LED, OUTPUT);
    pinMode(BUZZER, OUTPUT);
    
    digitalWrite(RED_LED, LOW);
    digitalWrite(GREEN_LED, HIGH);
    digitalWrite(BUZZER, LOW);

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

void loop() {
    int tempValue = analogRead(TEMP_SENSOR_PIN);  // Read temperature sensor
    int smokeValue = analogRead(SMOKE_SENSOR_PIN);  // Read smoke sensor
    
    // Convert potentiometer values (0-4095) to temperature & smoke levels
    float temperature = map(tempValue, 0, 4095, 20, 100);  // 20°C to 100°C
    float smokeLevel = map(smokeValue, 0, 4095, 0, 100);   // 0 to 100 (arbitrary scale)
    
    Serial.print("Temperature: "); Serial.print(temperature); Serial.println(" °C");
    Serial.print("Smoke Level: "); Serial.print(smokeLevel); Serial.println(" %");

    // Clear OLED display
    display.clearDisplay();
    display.setTextSize(1);
    display.setTextColor(WHITE);

    display.setCursor(10, 10);
    display.print("Temp: "); display.print(temperature); display.print(" C");
    
    display.setCursor(10, 30);
    display.print("Smoke: "); display.print(smokeLevel); display.print(" %");

    if (temperature > 60 || smokeLevel > 50) {  // Fire danger threshold
        digitalWrite(RED_LED, HIGH);
        digitalWrite(GREEN_LED, LOW);
        digitalWrite(BUZZER, HIGH);
        Serial.println("🔥 FIRE ALERT! 🔥");

        display.setCursor(10, 50);
        display.setTextSize(2);
        display.print("FIRE ALERT!");
    } else {
        digitalWrite(RED_LED, LOW);
        digitalWrite(GREEN_LED, HIGH);
        digitalWrite(BUZZER, LOW);

        display.setCursor(10, 50);
        display.print("SAFE");
    }

    // Show on OLED display
    display.display();
    delay(2000);
}