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

#define SCREEN_WIDTH 128  // OLED display width
#define SCREEN_HEIGHT 64  // OLED display height
#define OLED_RESET    -1  // Reset pin (or -1 if sharing Arduino reset pin)
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// WiFi credentials
const char* ssid     = "your_SSID";
const char* password = "your_PASSWORD";

// MQ-135 sensor pin
const int airQualityPin = 34; // Analog input pin

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

  // Initialize OLED display
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { // 0x3C is the I2C address for the OLED
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  display.clearDisplay();
  display.setTextSize(1);      
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);

  // Connect to Wi-Fi
  WiFi.begin(ssid, password);
  Serial.print("Connecting to WiFi...");
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("Connected!");
  Serial.print("IP Address: ");
  Serial.println(WiFi.localIP());

  // Display IP Address on OLED
  display.setCursor(0, 0);
  display.print("IP Address:");
  display.setCursor(0, 10);
  display.print(WiFi.localIP());
  display.display();
  delay(2000);
}

void loop() {
  // Read air quality value
  int airQualityValue = analogRead(airQualityPin);

  // Display air quality value on OLED
  display.clearDisplay();
  display.setCursor(0, 0);
  display.print("Air Quality:");
  display.setCursor(0, 10);
  display.print(airQualityValue);
  display.display();

  // Print air quality value to Serial Monitor
  Serial.print("Air Quality Value: ");
  Serial.println(airQualityValue);

  delay(2000); // Delay for readability and to avoid constant refresh
}