import network
from secrets import secrets
from utime import sleep
import machine
import ssd1306

# Inicializar pantalla SSD1306 (ajusta pines si es necesario)
i2c = machine.I2C(0, scl=machine.Pin(5), sda=machine.Pin(4))  # Para ESP32
oled = ssd1306.SSD1306_I2C(128, 64, i2c)

# Cargar datos de red
ssid = secrets['ssid']
password = secrets['password']

# Pantalla: Iniciando conexión
oled.fill(0)
oled.text('Conectando a:', 0, 0)
oled.text(ssid[:16], 0, 10)
oled.show()

# Activar módulo WiFi
wlan = network.WLAN(network.STA_IF)
wlan.active(True)
sleep(3)

oled.fill(0)
oled.text('Esperando WiFi...', 0, 0)
oled.show()
wlan.connect(ssid, password)
sleep(2)

# Verificar conexión
if wlan.isconnected():
    ip = wlan.ifconfig()[0]
    
    oled.fill(0)
    oled.text('¡Conexion exitosa!', 0, 0)
    oled.text('IP:', 0, 10)
    oled.text(ip, 0, 20)
    oled.text('Ping local...', 0, 35)
    oled.show()
    
    sleep(20)  # Esperar 20 segundos como en el código original

else:
    oled.fill(0)
    oled.text('¡ERROR!', 0, 0)
    oled.text('No hay conexion', 0, 10)
    oled.text('Verifica secrets.py', 0, 20)
    oled.show()