Stick man dances randomly to your favorite song (.wav)

dance.py

import numpy as np
import librosa
import time
import os
import pygame

steps =[
"""
 o
\ /
 |
/ \\
""", 
"""
 o
\ 
 |\\
/ \\
""", 
"""
 o
  /
/|
/ \\
""",
"""
 o
  /
/|
 \\ \\
""",
"""
 o
  /
/|
/ /
""",
"""
 o
\\ _
 | 
/ \\
"""
]

#  audio_path = "beats.wav"
audio_path = "corona.wav"
y, sr = librosa.core.load(audio_path)

tempo, beat_frames = librosa.beat.beat_track(y=y, sr=sr)
beat_times = librosa.frames_to_time(beat_frames, sr=sr)
dance_moves = np.random.choice(steps, len(beat_times))

i = 0

pygame.mixer.init()
pygame.mixer.music.load(audio_path)
pygame.mixer.music.play()
t1 = time.time()
while True:
    try:
        if time.time() - t1 >= beat_times[i]:
            os.system('clear')
            print(dance_moves[i])
            i+=1
    except:
        break