#define BLYNK_TEMPLATE_ID "TMPL3DWypFKeB"
#define BLYNK_TEMPLATE_NAME "Automated Water Level Monitoring"

#include <LiquidCrystal_I2C.h>
#include <Wire.h>
#include <WiFiClient.h>
#include <BlynkSimpleEsp32.h>

#define LED1 2
#define LED2 4
#define LED3 5
#define LED4 18
#define trig 12
#define echo 13
#define relay 14
#define LED5 14

int MaxLevel = 400;
int Level1 = 350;
int Level2 = 250;
int Level3 = 100;
int Level4 = 2;

LiquidCrystal_I2C lcd(0x27, 16, 2);

BlynkTimer timer;

char auth[] = "x4BEfU9D5jBkFkq6x-4NtgT2YlHSvAf3";
char ssid[] = "Wokwi-GUEST";
char pass[] = "";

int pumpState = HIGH;
int motorMode = 0; // 0 for normal, 1 for override

void setup() {
  Serial.begin(115200);
  Blynk.begin(auth, ssid, pass);
  lcd.init();
  lcd.backlight();
  pinMode(LED1, OUTPUT);
  pinMode(LED2, OUTPUT);
  pinMode(LED3, OUTPUT);
  pinMode(LED4, OUTPUT);
  pinMode(LED5, OUTPUT);
  pinMode(trig, OUTPUT);
  pinMode(echo, INPUT);
  pinMode(relay, OUTPUT);
  digitalWrite(relay, pumpState == LOW ? LOW : HIGH);
  lcd.setCursor(0, 0);
  lcd.print("System");
  lcd.setCursor(4, 1);
  lcd.print("Loading..");
  delay(4000);
  lcd.clear();
  Blynk.virtualWrite(V1, pumpState);
  Blynk.virtualWrite(V2, motorMode); // Initialize V2 (motor override switch) to OFF
}

void ultrasonic() {
  digitalWrite(trig, LOW);
  delayMicroseconds(4);
  digitalWrite(trig, HIGH);
  delayMicroseconds(10);
  digitalWrite(trig, LOW);
  long t = pulseIn(echo, HIGH);
  int distance = t / 29 / 2;

  Serial.println(distance);

  if (motorMode == 0) {
    // Normal mode, use ultrasonic sensor
    if (distance <= MaxLevel) {
      int blynkDistance = map(distance, 0, MaxLevel, 0, 100);
      Blynk.virtualWrite(V0, blynkDistance);
    } else {
      Blynk.virtualWrite(V0, 100);
    }
  } else {
    // Override mode, use manual control
    Blynk.virtualWrite(V0, pumpState ? 100 : 0);
  }

  lcd.setCursor(0, 0);
  lcd.print("WLevel:");

  if (Level1 <= distance) {
    lcd.setCursor(8, 0);
    lcd.print("Very Low");
    lcd.setCursor(0, 1);
    lcd.print("Motor is ON ");
    digitalWrite(LED1, HIGH);
    digitalWrite(LED2, LOW);
    digitalWrite(LED3, LOW);
    digitalWrite(LED4, LOW);
    digitalWrite(LED5, LOW);
    pumpState = HIGH;
  } else if (Level2 <= distance && Level1 > distance) {
    lcd.setCursor(8, 0);
    lcd.print("Low");
    lcd.print("      ");
    lcd.setCursor(0, 1);
    digitalWrite(LED1, HIGH);
    digitalWrite(LED2, HIGH);
    digitalWrite(LED3, LOW);
    digitalWrite(LED4, LOW);
    pumpState = HIGH;
  } else if (Level3 <= distance && Level2 > distance) {
    lcd.setCursor(8, 0);
    lcd.print("Medium");
    lcd.print("      ");
    lcd.setCursor(0, 1);
    digitalWrite(LED1, HIGH);
    digitalWrite(LED2, HIGH);
    digitalWrite(LED3, HIGH);
    digitalWrite(LED4, LOW);
    pumpState = HIGH;
  } else if (Level4 <= distance && Level3 > distance) {
    lcd.setCursor(8, 0);
    lcd.print("Full");
    lcd.print("      ");
    digitalWrite(LED1, HIGH);
    digitalWrite(LED2, HIGH);
    digitalWrite(LED3, HIGH);
    digitalWrite(LED4, HIGH);
    lcd.setCursor(0, 1);
    lcd.print("Motor is OFF");
    pumpState = LOW;
  } else if (distance < Level4) {
    lcd.setCursor(8, 0);
    lcd.print("Full");
    lcd.print("      ");
    lcd.setCursor(0, 1);
    lcd.print("Motor is OFF");
    digitalWrite(LED5, HIGH);
    pumpState = LOW;
  }

  Blynk.virtualWrite(V1, pumpState);
}

BLYNK_WRITE(V1) {
  int relayState = param.asInt();

  if (motorMode == 0 || motorMode == 1) {
    // In normal mode or override mode, update motor state
    if (relayState == HIGH && digitalRead(relay) == LOW) {
      digitalWrite(relay, HIGH);
      lcd.setCursor(0, 1);
      lcd.print("Motor is OFF");
      pumpState = LOW;
    } else if (relayState == LOW && digitalRead(relay) == HIGH) {
      digitalWrite(relay, LOW);
      lcd.setCursor(0, 1);
      lcd.print("Motor is ON ");
      pumpState = HIGH;
    }
  }
}

BLYNK_WRITE(V2) {
  int overrideMode = param.asInt();
  
  if (overrideMode == 1) {
    // Enable motor override mode
    motorMode = 1;
  } else {
    // Disable motor override mode and revert to normal
    motorMode = 0;
    Blynk.virtualWrite(V1, pumpState); // Sync V1 with the current motor state
  }
}

void loop() {
  ultrasonic();
  Blynk.run();
}