import network
import time
from machine import Pin
import dht
import ujson
from umqtt.simple import MQTTClient

# Assuming Blynk library is properly installed and supported for MicroPython
from blynklib_mp import Blynk

import machine
from machine import SoftI2C
from lcd_api import LcdApi
from i2c_lcd import I2cLcd

# WiFi credentials
WIFI_SSID = "Your_WiFi_SSID"
WIFI_PASSWORD = "Your_WiFi_Password"

# MQTT Server Parameters
MQTT_CLIENT_ID = ""
MQTT_BROKER = "broker.mqttdashboard.com"
MQTT_USER = ""
MQTT_PASSWORD = ""
MQTT_TOPIC = "wokwi-weather"

# Blynk parameters
BLYNK_AUTH_TOKEN = "Your_Blynk_Auth_Token"

# Initialize DHT22 sensor
sensor = dht.DHT22(Pin(15))

# Initialize I2C for LCD
I2C_ADDR = 0x27
totalRows = 2
totalColumns = 16
i2c = SoftI2C(scl=Pin(22), sda=Pin(21), freq=10000)
lcd = I2cLcd(i2c, I2C_ADDR, totalRows, totalColumns)

# Initialize Blynk
blynk = Blynk(BLYNK_AUTH_TOKEN)

# Connect to WiFi
print("Connecting to WiFi", end="")
sta_if = network.WLAN(network.STA_IF)
sta_if.active(True)
sta_if.connect(WIFI_SSID, WIFI_PASSWORD)
while not sta_if.isconnected():
    print(".", end="")
    time.sleep(0.1)
print(" Connected!")

# Main loop
prev_weather = ""
while True:
    # Measure weather conditions
    print("Measuring weather conditions... ", end="")
    sensor.measure()
    temp = sensor.temperature()
    humidity = sensor.humidity()
    
    # Construct JSON message for MQTT
    message = ujson.dumps({
        "temp": temp,
        "humidity": humidity,
    })
    
    # Publish to MQTT if there's a change in weather data
    if message != prev_weather:
        print("Updated!")
        print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))
        client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
        client.connect()
        client.publish(MQTT_TOPIC, message)
        client.disconnect()
        prev_weather = message
    else:
        print("No change")

    # Display on LCD
    lcd.putstr(message)
    lcd.clear()

    # Blynk virtual write - update Blynk app with latest sensor data
    blynk.virtual_write(0, temp)  # Virtual pin 0 for temperature
    blynk.virtual_write(1, humidity)  # Virtual pin 1 for humidity

    blynk.run()  # Run Blynk main loop

    time.sleep(1)  # Adjust interval as needed