Smart Urban Farming Powered by Aquaponics

import os, sys, io
import math
import M5
from M5 import *
from unit import AngleUnit
from m5espnow import M5ESPNow
import json
from hardware import Timer
import time
import network
from umqtt.simple import MQTTClient




home = None
Row1Label = None
Row2Label = None
option = None
label4 = None
paramLabel = None
valueLabel = None
plant1Label = None
plant2Label = None
plant3Label = None
plant4Label = None
SUFLabel = None
powerLabel = None
angle_0 = None
tableLabel = None
LEVEL = None
plant_profiles = None
angle = None
selectedPlant = None
selectedRow = None
selectedParam = None
flowRateLabel = None
co2LevelLabel = None
lightDurationLabel = None
lightIntensityLabel = None
humidityLabel = None
waterTempLabel = None
airTempLabel = None
clicked = False
timer0 = None
sendLabel = None
rotateRow=False

espnow_0 = None
mqtt_client = None
wlan = None
SSID = 'WIFI'
WLAN_PW = 'WIFI_PASSWORD'


# Row MAC addresses (for ESP-NOW communication)
ROW_MAC_LIST = ['2cbcbb81f7c0', '2cbcbb81F924']
MQTT = True #If MQTT is true, the master will send data to the mqtt server. if false, will only use espnow

def timer0_cb(t):
  global espnow_0, clicked, selectedPlant, selectedRow
  if clicked:
    # label0.setText(str('a'))
    clicked = False
    profile_to_data(selectedRow, selectedPlant)
  time.sleep_ms(10)



#Sends data from a plant profile dict, to send_data
#This is becuase old version had the ability to send/update single params, and keeping this setup allows for that to be reimplemented
def profile_to_data(row_number, plant_profile):
    """
    Translates the plant profile data to the dataAsList format.
    
    Args:
    plant_profile (dict): The plant profile containing the plant parameters.
    row_number (int): The row number to use for the MQTT topic.
    """
    send_data(
        row_number, 
        plant_profile['plantName'], 
        plant_profile['airTemp'], 
        plant_profile['waterTemp'], 
        plant_profile['humidity'], 
        plant_profile['lightIntensity'], 
        plant_profile['lightDuration'], 
        plant_profile['co2Level'], 
        plant_profile['flowRate']
    )

# Function to send data to a specific row
def send_data(rowNo, plantName, airTemp, waterTemp, humidity, lightIntensity, lightDuration, co2Level, flowRate):
    global espnow_0
    
    # Convert data to dict 
    rowData = {
        "plantName": plantName,
        "airTemp": airTemp,
        "waterTemp": waterTemp,
        "humidity": humidity,
        "lIntensity": lightIntensity,
        "lDuration": lightDuration,
        "co2Level": co2Level,
        "flowRate": flowRate,
    }

    try:
        #Convert to json
        json_data = json.dumps(rowData)
        payload = json_data.encode('utf-8')

        # Send the data to the selectedRow
        print(f"Sending to peer ID: {rowNo}, Data: {json_data}")
        espnow_0.send_data(int(rowNo), payload)
        print(f"Successfully sent to peer ID: {rowNo}")
    except Exception as e:
        print(f"Failed to send data: {e}")

    # MQTT Publishing (optional)
    if MQTT:
        time.sleep_ms(100)
        print("Mqtt send is on")
        if not wlan.isconnected():
            print("Wi-Fi not connected")
            return
        
        try:
            json_data = json.dumps(rowData)
            publish_with_retry(f'm5stack/row{rowNo}/dataList', json_data)
        except Exception as e:
            print(f"Publish error: {e}")

# MQTT Publishing with Retry
def publish_with_retry(topic, message, retries=3):
    global mqtt_client
    for attempt in range(retries):
        try:
            mqtt_client.publish(topic, message, qos=1)
            print(f"Published: {message} to {topic}")
            return
        except Exception as e:
            print(f"Publish failed ({attempt + 1}/{retries}): {e}")
            if not reconnect_mqtt():
                break
    print("Failed to publish message after retries")

# MQTT Reconnect Function
def reconnect_mqtt():
    global mqtt_client
    try:
        mqtt_client.connect()
        print("MQTT reconnected")
        return True
    except Exception as e:
        print(f"Failed to reconnect MQTT: {e}")
        return False

#hdies all labels exceot the ones passed as a list
def hide_except(labelsToKeep):
    all_labels = [sendLabel, Row1Label, Row2Label, plant1Label, plant2Label, plant3Label, plant4Label, paramLabel, valueLabel, option, SUFLabel, powerLabel, home, tableLabel, flowRateLabel, co2LevelLabel, lightDurationLabel, lightIntensityLabel, humidityLabel, waterTempLabel, airTempLabel]
    
    # Hide all labels
    for label in all_labels:
        label.setVisible(False)

    # Make labels in labelsToKeep visible
    for label in labelsToKeep:
        label.setVisible(True)

#Displays the selected plant profile using multiple labels.
def display_selected_plant(selected_plant_dict):
    if not selected_plant_dict:
        return

    # Create or update labels
    # plantNameLabel.setText(f"Plant: {selected_plant_dict['plantname']}")
    airTempLabel.setText(f"Air Temp: {selected_plant_dict['airTemp']} C")
    waterTempLabel.setText(f"Water Temp: {selected_plant_dict['waterTemp']} C")
    humidityLabel.setText(f"Humidity: {selected_plant_dict['humidity']}%")
    lightIntensityLabel.setText(f"Light Intensity: {selected_plant_dict['lightIntensity']}")
    lightDurationLabel.setText(f"Light Duration: {selected_plant_dict['lightDuration']} hrs")
    co2LevelLabel.setText(f"CO2 Level: {selected_plant_dict['co2Level']} ppm")
    flowRateLabel.setText(f"Flow Rate: {selected_plant_dict['flowRate']} L/min")

    # Position labels dynamically
    labels = [
        plantNameLabel, airTempLabel, waterTempLabel, humidityLabel,
        lightIntensityLabel, lightDurationLabel, co2LevelLabel, flowRateLabel
    ]

    for i, label in enumerate(labels):
        # label.setPosition(base_x, base_y + (i * spacing))
        label.setVisible(True)  # Ensure it's visible



#Hide all labels except home labels
def MAIN():
    global home, Row1Label, Row2Label, option, label4, paramLabel, valueLabel, plant1Label, plant2Label, plant3Label, plant4Label, SUFLabel, powerLabel, angle_0
    # home.setImage("res/img/bg.png")
    hide_except([SUFLabel, powerLabel, home])
    powerLabel.setText('Powered by Aquaponics')
    SUFLabel.setText('Smart Urban Farming')


def SEND():
  hide_except([sendLabel,flowRateLabel, co2LevelLabel, lightDurationLabel, lightIntensityLabel, humidityLabel, waterTempLabel, airTempLabel])
  display_selected_plant(selectedPlant)
  # Describe this function...
def Options():
    global rotateRow,LEVEL, plant_profiles, angle, selectedPlant, selectedRow, selectedParam, home, Row1Label, Row2Label, option, label4, paramLabel, valueLabel, plant1Label, plant2Label, plant3Label, plant4Label, SUFLabel, powerLabel, angle_0
    
    # Helper function to set label colors
    def set_plant_labels_color(selected_label, other_labels):
        selected_label.setColor(0x330033, 0xffffff)
        for label in other_labels:
            label.setColor(0xffffff, 0x990000)

    if LEVEL == 1:
        # Handle row label colors based on angle
        if angle < 45:
            rotateRow = True
            Row1Label.setColor(0xffffff, 0x5a2a2a)
            Row2Label.setColor(0xffffff, 0x5a2a2a)
        elif angle < 90:
            rotateRow = False

            selectedRow = 1
            Row1Label.setColor(0x330033, 0xffffff)
            Row2Label.setColor(0xffffff, 0x5a2a2a)
        else:
            rotateRow = False
            selectedRow = 2
            Row1Label.setColor(0xffffff, 0x5a2a2a)
            Row2Label.setColor(0x330033, 0xffffff)

    elif LEVEL == 2:
        Row1Label.setVisible(False)
        Row2Label.setVisible(False)

        # Set plant labels based on angle and level
        if angle < 45:
            selectedPlant = plant_profiles['Lettuce']
            set_plant_labels_color(plant1Label, [plant2Label, plant3Label, plant4Label])
        elif angle < 90:
            selectedPlant = plant_profiles['Basil']
            set_plant_labels_color(plant2Label, [plant1Label, plant3Label, plant4Label])
        elif angle < 135:
            selectedPlant = plant_profiles['Strawberry']
            set_plant_labels_color(plant3Label, [plant1Label, plant2Label, plant4Label])
        else:
            selectedPlant = plant_profiles['CUSTOM']
            set_plant_labels_color(plant4Label, [plant1Label, plant2Label, plant3Label])

        option.setText(str(selectedRow))

    elif LEVEL == 3:
        # Set parameters based on angle
        params = [
            ('airTemp', 'Air Temp', selectedPlant['airTemp']),
            ('waterTemp', 'Water Temp', selectedPlant['waterTemp']),
            ('humidity', 'Humidity', selectedPlant['humidity']),
            ('lightIntensity', 'Light Intensity', selectedPlant['lightIntensity']),
            ('lightDuration', 'Light Duration', selectedPlant['lightDuration']),
            ('co2Level', 'CO2 Level', selectedPlant['co2Level']),
            ('flowRate', 'Flow Rate', selectedPlant['flowRate']),
        ]
        for i, (param, label, value) in enumerate(params):
            if angle < (i + 1) * 25:
                selectedParam = param
                paramLabel.setText(label)
                valueLabel.setText(str(value))
                break

        option.setText(str(selectedPlant['plantName']))

    elif LEVEL == 4:
        # Update plant values based on selected parameter
        param_updates = {
            'airTemp': 10 + round(angle / 8),
            'waterTemp': 5 + round(angle / 8),
            'humidity': round(100 * (angle / 180)),
            'lightIntensity': round(angle / 0.09),
            'lightDuration': 10 + round(angle / 90),
            'co2Level': 300 + round(900 * (angle / 180)),
            'flowRate': round(angle / 18)
        }

        if selectedParam in param_updates:
            selectedPlant[selectedParam] = param_updates[selectedParam]
            valueLabel.setText(str(selectedPlant[selectedParam]))

# Describe this function...
def ROW():
    # Use hide_except function to only show relevant labels
    hide_except([Row1Label, Row2Label])

    # Set row labels text
    Row1Label.setText('Row1')
    Row2Label.setText('Row2')

# Describe this function...
def PLANT():
    # Use hide_except function to only show relevant labels
    hide_except([option, plant1Label, plant2Label, plant3Label, plant4Label])

    # Set plant labels text
    plant1Label.setText('Lettuce')
    plant2Label.setText('Basil')
    plant3Label.setText('Strawberry')
    plant4Label.setText('CUSTOM')


# Describe this function...
def PARAM():
    hide_except([option, paramLabel, valueLabel])

def btnA_wasPressed_event(state):
    global LEVEL, rotateRow, SUFLabel

    # Increment LEVEL, reset to 0 if it exceeds 4
    LEVEL += 1

    # Call appropriate function based on LEVEL
    if LEVEL == 1:
        ROW()
    elif LEVEL == 2:
      if rotateRow:
        LEVEL = 0
        hide_except([SUFLabel])
        for i in range(10):
          SUFLabel.setText("Rotating Row...")
          time.sleep_ms(500)
          SUFLabel.setText("")
          time.sleep_ms(500)
        SUFLabel("Smart Urban Farming")
        MAIN()
      else:
        PLANT()
    elif LEVEL == 3:
        PARAM()
    elif LEVEL == 5:
      SEND()
    elif LEVEL > 5:  # LEVEL > 4
        LEVEL = 0
        MAIN()
def btnB_wasPressed_event(state):
  global clicked
  print("beans")
  if LEVEL == 5:
    clicked = True

def btnC_wasPressed_event(state):
    global LEVEL

    # Decrement LEVEL, reset to 0 if it goes below 1
    LEVEL -= 1
    if LEVEL == 1:
        ROW()
    elif LEVEL == 2:
        PLANT()
    elif LEVEL == 3:
        PARAM()
    elif LEVEL == 4:
      PARAM()
    elif LEVEL ==5:
      SEND()
    else:
        LEVEL = 0
        MAIN()



def setup():
  global sendLabel, plantNameLabel, airTempLabel, waterTempLabel, humidityLabel, espnow_0, timer0,wlan, mqtt_client
  global lightIntensityLabel, lightDurationLabel, co2LevelLabel, flowRateLabel
  global tableLabel, home, Row1Label, Row2Label, option, label4, paramLabel, valueLabel, plant1Label, plant2Label, plant3Label, plant4Label, SUFLabel, powerLabel, angle_0, LEVEL, plant_profiles, angle, selectedPlant, selectedRow, selectedParam

  if MQTT:
    wlan = network.WLAN(network.STA_IF)
    wlan.active(True)
    wlan.connect(SSID, WLAN_PW)
 

  M5.begin()
  Widgets.fillScreen(0x5a2a2a)

  # Initialize widgets
  home = Widgets.Image("res/img/default.png", 0, -30, scale_x=.4, scale_y=.4)
  Row1Label = Widgets.Label("Row1", 131, 71, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  Row2Label = Widgets.Label("Row2", 131, 108, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  option = Widgets.Label("label0", 0, 0, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu24)
  label4 = Widgets.Label("label4", 282, 215, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  paramLabel = Widgets.Label("Param:", 85, 63, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu24)
  valueLabel = Widgets.Label("value", 134, 121, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  plant1Label = Widgets.Label("plant1", 120, 50, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  plant2Label = Widgets.Label("pant2", 135, 100, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  plant3Label = Widgets.Label("plant3", 105, 145, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  plant4Label = Widgets.Label("plant4", 127, 191, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  SUFLabel = Widgets.Label("Smart Urban Farming", 22, 94, 1.0, 0xffffff, 0x54ca2c, Widgets.FONTS.DejaVu24)
  powerLabel = Widgets.Label("Powered by Aquaponics", 46, 121, 1.0, 0xffffff, 0x32637e, Widgets.FONTS.DejaVu18)
  tableLabel = Widgets.Label("Table", 0,0,1.0,0xffffff, 0x32637e, Widgets.FONTS.DejaVu18)
  plantNameLabel = Widgets.Label("", 20, 50, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  airTempLabel = Widgets.Label("", 20, 25, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  waterTempLabel = Widgets.Label("", 20, 50, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  humidityLabel = Widgets.Label("", 20, 75, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  lightIntensityLabel = Widgets.Label("", 20, 100, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  lightDurationLabel = Widgets.Label("", 20, 125, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  co2LevelLabel = Widgets.Label("", 20, 150, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  flowRateLabel = Widgets.Label("", 20, 175, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)
  sendLabel = Widgets.Label("send", 133, 214, 1.0, 0xffffff, 0x5a2a2a, Widgets.FONTS.DejaVu18)

  
  if MQTT:
    while not wlan.isconnected():
        print("Connecting to Wi-Fi...")
        time.sleep(1)
    print("Connected to Wi-Fi")
    
    try:
        mqtt_client = MQTTClient('TacoMaster', 'test.mosquitto.org', port=1883, user='', password='', keepalive=60)
        mqtt_client.connect()
    except Exception as e:
        print(f"MQTT connection error: {str(e)}")


  # Set button callbacks
  BtnA.setCallback(type=BtnA.CB_TYPE.WAS_PRESSED, cb=btnA_wasPressed_event)
  BtnB.setCallback(type=BtnB.CB_TYPE.WAS_PRESSED, cb=btnB_wasPressed_event)
  BtnC.setCallback(type=BtnC.CB_TYPE.WAS_PRESSED, cb=btnC_wasPressed_event)

  espnow_0 = M5ESPNow(0)
  for index, mac in enumerate(ROW_MAC_LIST):
      espnow_0.set_add_peer(mac, (index + 1), 0, False)
      print(f"Peer added: {mac} with ID: {index + 1}")

  print(f"Peer added: 2cbcbb81f7c0 with ID: 1")


  # Initialize plant profiles
  plant_profiles = {
    'Lettuce': {'plantName': 'Lettuce', 'airTemp': 22, 'waterTemp': 18, 'humidity': 70, 'lightIntensity': 400, 'lightDuration': 16, 'co2Level': 400, 'flowRate': 1},
    'Basil': {'plantName': 'Basil', 'airTemp': 25, 'waterTemp': 22, 'humidity': 65, 'lightIntensity': 500, 'lightDuration': 14, 'co2Level': 450, 'flowRate': 1.2},
    'Strawberry': {'plantName': 'Strawberry', 'airTemp': 20, 'waterTemp': 19, 'humidity': 75, 'lightIntensity': 350, 'lightDuration': 12, 'co2Level': 350, 'flowRate': 1.5},
    'CUSTOM': {'plantName': 'CUSTOM', 'airTemp': 0, 'waterTemp': 0, 'humidity': 0, 'lightIntensity': 0, 'lightDuration': 0, 'co2Level': 0, 'flowRate': 0}
  }
  timer0 = Timer(0)

  timer0.init(mode=Timer.PERIODIC, period=500, callback=timer0_cb)

  LEVEL = 0
  angle_0 = AngleUnit((36, 26))


  # Start the main process
  MAIN()



def loop():
  global home, Row1Label, Row2Label, option, label4, paramLabel, valueLabel, plant1Label, plant2Label, plant3Label, plant4Label, SUFLabel, powerLabel, angle_0, LEVEL, plant_profiles, angle, selectedPlant, selectedRow, selectedParam
  
  M5.update()
  # get to roughly between 0 and 180
  angle = (angle_0.get_value()) / 364
  label4.setText(str(LEVEL))
  Options()


if __name__ == '__main__':
  try:
    setup()
    while True:
      loop()
  except (Exception, KeyboardInterrupt) as e:
    try:
      from utility import print_error_msg
      print_error_msg(e)
    except ImportError:
      print("please update to latest firmware")