from machine import Pin, I2C, PWM
import ssd1306

# ESP32 Pin assignment
i2c = I2C(0, scl=Pin(22), sda=Pin(21))

# OLED Display
oled = ssd1306.SSD1306_I2C(128, 64, i2c, addr=0x3C)  # Usa la dirección que encontraste

# Clear the display
oled.fill(0)
oled.show()

# Pin Definitions
INPUT_PIN_1 = Pin(23, Pin.IN)
INPUT_PIN_2 = Pin(25, Pin.IN)
INPUT_PIN_3 = Pin(26, Pin.IN)
VACUUM_PUMP_PIN = Pin(5, Pin.OUT)
VACUUM_VALVE_PIN = Pin(18, Pin.OUT)
SEAL_VALVE_PIN = Pin(19, Pin.OUT)

# Initialize Pins
VACUUM_PUMP_PIN.value(0)
VACUUM_VALVE_PIN.value(0)
SEAL_VALVE_PIN.value(0)

# Display initial status on OLED
oled.fill(0)
oled.text('Init Complete', 10, 0)
oled.text('Input 1: {}'.format(INPUT_PIN_1.value()), 0, 10)
oled.text('Input 2: {}'.format(INPUT_PIN_2.value()), 0, 20)
oled.text('Input 3: {}'.format(INPUT_PIN_3.value()), 0, 30)
oled.text('Vac Pump: {}'.format(VACUUM_PUMP_PIN.value()), 0, 40)
oled.text('Vac Valve: {}'.format(VACUUM_VALVE_PIN.value()), 0, 50)
oled.text('Seal Valve: {}'.format(SEAL_VALVE_PIN.value()), 0, 60)
oled.show()