import cv2, mediapipe as mp, serial, time

ser = serial.Serial('COM4', 9600)
mpHands = mp.solutions.hands
hands = mpHands.Hands()
cap = cv2.VideoCapture(0)

while True:
    success, img = cap.read()
    imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    results = hands.process(imgRGB)
    fingers = 0

    if results.multi_hand_landmarks:
        for hand in results.multi_hand_landmarks:
            landmarks = hand.landmark
            # просте визначення піднятих пальців
            for tip in [8, 12, 16, 20]:
                if landmarks[tip].y < landmarks[tip-2].y:
                    fingers += 1
            if landmarks[4].x < landmarks[3].x:
                fingers += 1

    ser.write(f"{fingers}\n".encode())
    cv2.imshow("Camera", img)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

int leds[] = {2, 3, 4, 5, 6};
#define CLK 8
#define DT 9
#define SWITCH_PIN 7

int lastStateCLK;
int brightness = 120;
int fingerCount = 0;
int lastValidCount = 0;
unsigned long lastDataTime = 0;
const unsigned long DATA_TIMEOUT = 500;

void setup() {
  Serial.begin(9600);
  for (int i = 0; i < 5; i++) pinMode(leds[i], OUTPUT);
  pinMode(CLK, INPUT);
  pinMode(DT, INPUT);
  pinMode(SWITCH_PIN, INPUT);
  lastStateCLK = digitalRead(CLK);
}

void loop() {
  if (Serial.available()) {
    int incoming = Serial.parseInt();
    if (incoming >= 1 && incoming <= 5) { fingerCount = incoming; lastValidCount = incoming; }
    else if (incoming == 0) fingerCount = 0;
    lastDataTime = millis();
  }
  if (millis() - lastDataTime > DATA_TIMEOUT && fingerCount != 0) fingerCount = lastValidCount;

  int currentStateCLK = digitalRead(CLK);
  if (currentStateCLK != lastStateCLK && currentStateCLK == HIGH) {
    if (digitalRead(DT) != currentStateCLK) brightness += 10;
    else brightness -= 10;
    brightness = constrain(brightness, 0, 255);
  }
  lastStateCLK = currentStateCLK;

  bool powerOn = digitalRead(SWITCH_PIN);
  if (powerOn) {
    for (int i = 0; i < 5; i++) {
      if (i < fingerCount) analogWrite(leds[i], brightness);
      else analogWrite(leds[i], 0);
    }
  } else for (int i = 0; i < 5; i++) analogWrite(leds[i], 0);
}