#include <Adafruit_NeoPixel.h>

Adafruit_NeoPixel a(4, 2, NEO_GRB + NEO_KHZ800); 
bool stav = true;

void setup() {
  a.begin();
  Serial.begin(9600);
  randomSeed(analogRead(0));
}

void loop() {
  if (Serial.available()) {
    Serial.read(); 
    stav = true;
  }

  Serial.println("aka nedela?");
  
  while (!Serial.available()) { 
  } 

  int nedela = Serial.parseInt(); 
  Serial.println(nedela);

  if (nedela >= 1 && nedela <= 4) { 
    a.clear();
    a.show();
    while (stav == true) {
      int rand = random(0,3); 

      for (int i = 0; i < nedela; i++) {
        if (rand == 0) {
          a.setPixelColor(i, a.Color(255, 0, 0)); 
          a.show();
        }
        else if (rand == 1) {
          a.setPixelColor(i, a.Color(255, 165, 0));
          a.show(); 
        }
        else if (rand == 2) {
          a.setPixelColor(i, a.Color(255, 255, 0)); 
          a.show();
        }
      }
      
      delay(50); 

      if (Serial.available()) { 
        stav = false;
      }
    }
  } else {
    Serial.println("od 1 do 4");
  }
}