import os
import time
import random
import RPi.GPIO as GPIO

GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)

LEDRed = 18
LEDYellow = 23
LEDGreen = 24
ButtonPin = 25

GPIO.setup(LEDRed, GPIO.OUT)
GPIO.setup(LEDYellow, GPIO.OUT)
GPIO.setup(LEDGreen, GPIO.OUT)
GPIO.setup(ButtonPin, GPIO.IN)

correct_button = 0

start = time.time()

while correct_button <= 10:
    led_choice = random.randint(1,3)
    GPIO.output(led, GPIO.HIGH)
    button_seen = False
    for i in range(200):
        if led_choice == 1:
            led = LEDRed
        if led_choice == 2:
            led = LEDYellow
            if GPIO.input(ButtonPin) == False:
                button_seen = True
        if led_choice == 3:
            led = LEDGreen
        time.sleep(0.001)
    if button_seen:
        correct_button = correct_button + 1
    GPIO.output(led, GPIO.LOW)

end = time.time()

time_taken = end - start
time_taken = str(time_taken)

print("Time taken in seconds is " + time_taken)

GPIO.cleanup()