Цифровое сердце

from tkinter import *
w = Tk()
w.geometry('230x245')
 
l1 = Label(text='',bg='white', width=10,height=5)
l2 = Label(text='',bg='white', width=10,height=5)
l3 = Label(text='',bg='white', width=10,height=5)
l4 = Label(text='',bg='white', width=10,height=5)
l5 = Label(text='',bg='red', width=10,height=5)
l6 = Label(text='',bg='white', width=10,height=5)
l7 = Label(text='',bg='white', width=10,height=5)
l8 = Label(text='',bg='white', width=10,height=5)
l9 = Label(text='',bg='white', width=10,height=5)
 
 
l1.grid(row=0, column=0)
l2.grid(row=0, column=1)
l3.grid(row=0, column=2)
l4.grid(row=1, column=0)
l5.grid(row=1, column=1)
l6.grid(row=1, column=2)
l7.grid(row=2, column=0)
l8.grid(row=2, column=1)
l9.grid(row=2, column=2)
 
print('Перемещайте элемент клавишами со стрелками вверх, вниз, вправо, влево')
 
def move_red(e):
    try:
        all_cells = w.winfo_children()
        red_cell = None
        grid_size = w.grid_size()     
        for i in all_cells:    
            if i.cget('bg') == 'red':
                red_cell = i
                break
        index_of_red = all_cells.index(i)    
        l = len(all_cells)
        elem_next = None
        col = None
        if e.keysym == 'Up':
            if red_cell.grid_info()['row'] == 0:
                return            
            elem_next = red_cell.grid_info()['row'] - 1            
            col = red_cell.grid_info()['column']            
        elif e.keysym == 'Down':
            elem_next = red_cell.grid_info()['row'] + 1
            col = red_cell.grid_info()['column']
        elif e.keysym == 'Left':
            if red_cell.grid_info()['column'] == 0:
                return
            elem_next = red_cell.grid_info()['row']        
            col = red_cell.grid_info()['column'] - 1
        elif e.keysym == 'Right':            
            if red_cell.grid_info()['column'] == grid_size[1] - 1:
                return            
            elem_next = red_cell.grid_info()['row']        
            col = red_cell.grid_info()['column'] + 1
        else:
            return 
        all_cells[elem_next * grid_size[1] + col].config(bg='red') 
        all_cells[index_of_red].config(bg='white')
    except IndexError:
        print('Не туды жмякаешь, тупица')   
 
w.bind('<KeyPress>', move_red)
w.mainloop()