drawEyes

WHITE = (255, 255, 255) # Python's color code for white
EYE_WIDTH = 3
EYE_HEIGHT = 3

def draw_snake_eyes(self, direction):
    pt = self.snake[0]

    if direction == 'UP':
        eye_pos1 = (pt.x + 5, pt.y + 5)
        eye_pos2 = (pt.x + BLOCK_SIZE - 10, pt.y + 5)
    elif direction == 'DOWN':
        eye_pos1 = (pt.x + 5, pt.y + BLOCK_SIZE - 10)
        eye_pos2 = (pt.x + BLOCK_SIZE - 10, pt.y + BLOCK_SIZE - 10)
    elif direction == 'LEFT':
        eye_pos1 = (pt.x + 5, pt.y + 5)
        eye_pos2 = (pt.x + 5, pt.y + BLOCK_SIZE - 10)
    elif direction == 'RIGHT':
        eye_pos1 = (pt.x + BLOCK_SIZE - 10, pt.y + 5)
        eye_pos2 = (pt.x + BLOCK_SIZE - 10, pt.y + BLOCK_SIZE - 10)

    # Draw eyes
    pygame.draw.rect(self.display, WHITE, pygame.Rect(*eye_pos1, EYE_WIDTH, EYE_HEIGHT))
    pygame.draw.rect(self.display, WHITE, pygame.Rect(*eye_pos2, EYE_WIDTH, EYE_HEIGHT))