import time
from seven_segment import pins_setup, display, clear, all_on, get_segments

# Pin configuration for 7-segment display
my_pins = [3, 4, 5, 6, 7, 2, 1]  # A, B, C, D, E, F, G
pins_setup(my_pins)  # Initialize display with pins

# Get segment objects for individual control
seg_A, seg_B, seg_C, seg_D, seg_E, seg_F, seg_G = get_segments()

delay = 1

def milkshake():
    seg_A.on()
    time.sleep(delay)
    seg_A.off()
    time.sleep(delay)
    seg_B.on()
    time.sleep(delay)
    seg_B.off()
    time.sleep(delay)
    seg_C.on()
    time.sleep(delay)
    seg_C.off()
    time.sleep(delay)
    seg_D.on()
    time.sleep(delay)
    seg_D.off()
    time.sleep(delay)
    seg_E.on()
    time.sleep(delay)
    seg_E.off()
    time.sleep(delay)
    seg_F.on()
    time.sleep(delay)
    seg_F.off()
    time.sleep(delay)
    seg_G.on()
    time.sleep(delay)
    seg_G.off()

while True:

    milkshake()
    
    display('9')
    time.sleep(1)

    display('8')
    time.sleep(1)

    display('7')
    time.sleep(1)

    clear()