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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

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

const int inputPins[] = {2, 3, 4, 5}; // Pins for input states
const int outputPins[] = {6, 7, 8, 9}; // Pins for output states
const int inputLedPins[] = {10, 11, 12, 13}; // Pins for input state LEDs
const int outputLedPins[] = {A0, A1, A2, A3}; // Pins for output state LEDs

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

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

  delay(2000);
  display.clearDisplay();
  display.display();
}

void loop() {

  int inputState[4];
  for (int i = 0; i < 4; ++i) {
    inputState[i] = digitalRead(inputPins[i]);
    digitalWrite(inputLedPins[i], inputState[i]);
  }

  int outputState[4];
  for (int i = 0; i < 4; ++i) {
    outputState[i] = digitalRead(outputPins[i]);
    digitalWrite(outputLedPins[i], outputState[i]);
  }

  // Display the state on OLED (same as before)
  display.clearDisplay();
  display.setTextSize(2.5);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(3, 10);
  display.print("I- ");
  for (int i = 0; i < 4; ++i) {
    display.print(inputState[i]);
    display.print(" ");
  }

  display.setCursor(3, 35);
  display.print("O- ");
  for (int i = 0; i < 4; ++i) {
    display.print(outputState[i]);
    display.print(" ");
  }

  display.display();
  delay(1000);
}