import machine
import network
import ssd1306
import utime
import urequests
import json

# Configuración del módulo Wi-Fi 
esp8266 = machine.UART(0, baudrate=115200, tx=Pin(0), rx=Pin(1))
esp8266.init(115200, tx=Pin(0), rx=Pin(1))
esp8266.write("AT+UART=115200,8,1,0,0\r\n")
esp8266.write("AT+CWMODE=1\r\n")
esp8266.write("AT+CWJAP=\"Tu_SSID\",\"Tu_Contraseña\"\r\n")  # Reemplaza con tu SSID y contraseña

# Configuración de la pantalla OLED
i2c = I2C(0, sda=Pin(2), scl=Pin(3), freq=400000)
oled = ssd1306.SSD1306_I2C(128, 64, i2c)

# URL de la API de OpenAI
API_URL = "https://api.openai.com/v1/chat/completions"

def send_message(message):
    headers = {
        "Authorization": "Bearer YOUR_API_KEY",  # Reemplaza con tu clave de API
        "Content-Type": "application/json"
    }

    payload = {
        "messages": [
            {"role": "system", "content": "You are Bing."},
            {"role": "user", "content": message}
        ]
    }

    response = urequests.post(API_URL, headers=headers, json=payload)

    if response.status_code == 200:
        return response.json()["choices"][0]["message"]["content"]
    else:
        return None

def display_message(message):
    oled.fill(0)
    oled.text("Chat Gpt dice:", 0, 0)
    oled.text(message, 0, 20)
    oled.show()

while True:
    if "WIFI GOT IP" in esp8266.read():
        print("Conexión Wi-Fi establecida")

    button_state = machine.Pin(4, machine.Pin.IN, machine.Pin.PULL_UP).value()

    if button_state == 0:
        response = send_message("Hola")
        if response:
            display_message(response)

    utime.sleep(0.1)