import csv
import itertools

if __name__ == "__main__":

    fp = open("/app/suno/data/audio_2ch_48khz_lg/music_tr.csv", "r")
    reader = csv.reader(fp)
    cycle_reader = itertools.cycle(reader)

    start = ""
    count = 0
    for idx, line in enumerate(cycle_reader):
        if idx == 0:
            start = line[0]

        if line[0] == start:
            count += 1
            print(f"restarting {idx} {count} {line[0]}")