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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

#define OLED_RESET -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

const int buttonPin = 2;  // Replace D2 with the pin connected to your button

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

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

  // Initialize button pin
  pinMode(buttonPin, INPUT);

  // Display initial message
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  display.print("Press the button");
  display.display();
}

void loop() {
  // Check if the button is pressed
  if (digitalRead(buttonPin) == HIGH) {

display.setTextColor(WHITE); 
  display.setCursor(0,10);
  display.println("STATE...      ");
    delay(2000);
  display.display();
  //display.clearDisplay();


 display.setTextColor(BLACK, WHITE); 
  display.setCursor(0,20);
  display.println("ON         ");
  display.display();
  display.setTextColor(WHITE); // 'inverted' text
  display.setCursor(0,30);
  display.println("OFF         ");
  display.display();
  delay(2000);


display.clearDisplay();

display.setTextColor(WHITE); 
  display.setCursor(0,10);
  display.println("STATE...      ");

display.setTextColor(WHITE); 
  // 'inverted' text
  display.setCursor(0,20);
  display.println("ON         ");
  display.display();
  display.setTextColor(BLACK, WHITE); 
  display.setCursor(0,30);
  display.println("OFF         ");
  display.display();

    // Clear the display
    display.clearDisplay();
  }
}