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

#define POT_PIN A0  
#define BUZZER_PIN 8

Adafruit_SSD1306 display(128, 64, &Wire, -1);

void setup() {
  pinMode(POT_PIN, INPUT);
  pinMode(BUZZER_PIN, OUTPUT);
 
  Serial.begin(9600);

  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }
  display.clearDisplay();
}

void loop() {
  int sensorValue = analogRead(POT_PIN);
  Serial.print("Air Quality: ");
  Serial.println(sensorValue);
 
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(WHITE);
  display.setCursor(10, 20);
  display.print("AQI: ");
  display.print(sensorValue);
  display.display();

  if (sensorValue > 600) { 
    digitalWrite(BUZZER_PIN, HIGH);
  } else {
    digitalWrite(BUZZER_PIN, LOW);
  }

  delay(1000);
}