Simulação de Luz

import tkinter as tk

class luzsimulador:
    def __init__(self, root):
        self.root = root
        self.root.title("Simulador de Luz")

        self.luz_amarela1 = tk.Canvas(root, width=50, height=50, bg="yellow")
        self.luz_amarela1.grid(row=0, column=2, padx=5, pady=5)

        self.luz_amarela2 = tk.Canvas(root, width=50, height=50, bg="gray")
        self.luz_amarela2.grid(row=1, column=2, padx=5, pady=5)

        self.luz_amarela3 = tk.Canvas(root, width=50, height=50, bg="gray")
        self.luz_amarela3.grid(row=2, column=2, padx=5, pady=5)

        self.luz_amarela4 = tk.Canvas(root, width=50, height=50, bg="gray")
        self.luz_amarela4.grid(row=3, column=2, padx=5, pady=5)

        self.estado_atual = 0
        self.mudar_estado()

    def mudar_estado(self):
        if self.estado_atual == 0:
            self.luz_amarela1.configure(bg="yellow")
            self.luz_amarela2.configure(bg="gray")
            self.luz_amarela3.configure(bg="gray")
            self.luz_amarela4.configure(bg="gray")
            self.estado_atual = 1
        elif self.estado_atual == 1:
            self.luz_amarela1.configure(bg="gray")
            self.luz_amarela2.configure(bg="yellow")
            self.luz_amarela3.configure(bg="gray")
            self.luz_amarela4.configure(bg="gray")
            self.estado_atual = 2
        elif self.estado_atual == 2:  # Luz verde
            self.luz_amarela1.configure(bg="gray")
            self.luz_amarela2.configure(bg="gray")
            self.luz_amarela3.configure(bg="yellow")
            self.luz_amarela4.configure(bg="gray")
            self.estado_atual = 3
        elif self.estado_atual == 3:
            self.luz_amarela1.configure(bg="gray")
            self.luz_amarela2.configure(bg="gray")
            self.luz_amarela3.configure(bg="gray")
            self.luz_amarela4.configure(bg="yellow")
            self.estado_atual = 0
        self.root.after(1000, self.mudar_estado)
if __name__ == "__main__":
    root = tk.Tk()
    root.geometry("300x300")
    simulador = luzsimulador(root)
    root.mainloop()