from machine import Pin, I2C
import ssd1306
import utime
import urequests
import json  # Importa la biblioteca json

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

# Configuración del botón
BUTTON_PIN = Pin(2, Pin.IN, Pin.PULL_UP)

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

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

    payload = {
        "messages": [
            {"Chistes": "Nutricion", "contenido"},
            {"Chistes": "Nutricion", "contenido": message}
        ]
    }

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

    if response.status_code == 200:
        return response.json()["Dudas"][0]["mensajes"]["contenido"]
    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:
    button_state = BUTTON_PIN.value()

    if button_state == 0:
        # El botón está presionado. Realiza alguna acción aquí.
        response = send_message("Hola")  # Envía un mensaje a la API de OpenAI
        if response:
            display_message(response)  # Muestra la respuesta en la pantalla OLED

    utime.sleep(0.1)