from machine import Pin 
from machine import Pin, I2C
import machine
import ssd1306 
import dht
import time

#ChatGPT
import network
import urequests
import json 
wlan = network.WLAN(network.STA_IF)
wlan.active(True)
wlan.connect("Wokwi-GUEST","")
while not wlan.isconnected():
    pass

def scroll(oled, text, char_width=6):
    text_width = len(text) * char_width
    for i in range(0, text_width + 128, 10):  # Ajusta el valor para controlar la velocidad
        oled.fill(0)
        oled.text(text, -i, 10)
        oled.show()
        time.sleep_ms(50)  # Ajusta según sea necesario

i2c = I2C(scl=Pin(22), sda=Pin(21))   
oled = ssd1306.SSD1306_I2C(128, 64, i2c, 0x3c)

p15 = Pin(27, Pin.IN)     
d=dht.DHT22(p15)

while True:
  d.measure()   
  t=d.temperature()    
  h=d.humidity()       
  print('Temperature =', t, 'C', 'Humidity =', h, '%')   
  time.sleep(1)      
  oled.fill(0)
  oled.text("Temperature", 20, 10)  
  oled.text(str(t), 40, 20)
  oled.text("C", 60, 20)
  oled.text("Humidity", 30, 40)    
  oled.text(str(h), 40, 55)
  oled.text("%", 60, 55)
  oled.show()

  sTemp = str(t)
  sHum = str(h)
  # SAY THIS IS A TEST!!!
  headers = {'Authorization': 'Bearer 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','Content-Type': 'application/json',}
  data = '{"model": "gpt-3.5-turbo","messages":[{"role": "user", "content": "En base a la temperatura: '+sTemp+' y esta humedad: '+ sHum +' que prevenciones recomiendas al manejar? dimelo en menos de 20 palabras\\n"}],"temperature": 0.5}'
  URL = "https://api.openai.com/v1/chat/completions"
  r = urequests.post(URL,headers=headers,data=data)
  person_dict = json.loads(r.text)
  print("\ngpt-3.5-turbo:")
  mensaje1 = person_dict['choices'][0]['message']['content']
  print(mensaje1)

  # SAY THIS IS A TEST!!!
  headers = {'Authorization': 'Bearer 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','Content-Type': 'application/json',}
  data = '{"model": "gpt-3.5-turbo-16k","messages":[{"role": "user", "content": "En base a la temperatura: '+sTemp+' y esta humedad: '+ sHum +' que prevenciones recomiendas al manejar? dimelo en menos de 25 palabras\\n"}],"temperature": 0.5}'
  URL = "https://api.openai.com/v1/chat/completions"
  r = urequests.post(URL,headers=headers,data=data)
  person_dict = json.loads(r.text)
  print("\ngpt-3.5-turbo-16k:")
  mensaje2 = person_dict['choices'][0]['message']['content']
  print(mensaje2)

  # SAY THIS IS A TEST!!!
  headers = {'Authorization': 'Bearer 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','Content-Type': 'application/json',}
  data = '{"model": "gpt-3.5-turbo-1106","messages":[{"role": "user", "content": "En base a la temperatura: '+sTemp+' y esta humedad: '+ sHum +' que prevenciones recomiendas al manejar? dimelo en menos de 18 palabras\\n"}],"temperature": 0.5}'
  URL = "https://api.openai.com/v1/chat/completions"
  r = urequests.post(URL,headers=headers,data=data)
  person_dict = json.loads(r.text)
  print("\ngpt-3.5-turbo-1106:")
  mensaje3 = person_dict['choices'][0]['message']['content']
  print(mensaje3)
  # Clear the OLED display
  oled.fill(0)

  # Scroll the prevention tips on the OLED display
  scroll(oled, mensaje3)