import os
import librosa
import soundfile as sf

def get_slices(file_path, sample_rate=44100, slice_duration=10, output_dir='slices'):
    os.makedirs(output_dir, exist_ok=True)

    audio, sr = librosa.load(file_path, sr=sample_rate)
    slice_samples = slice_duration * sample_rate

    existing_slices = [f for f in os.listdir(output_dir) if f.endswith('.wav')]
    if existing_slices:
        max_index = max(int(f.split('_')[1].split('.')[0]) for f in existing_slices)
        start_index = max_index + 1
    else:
        start_index = 0

    num_slices = len(audio) // slice_samples
    for i in range(num_slices):
        start_sample = i * slice_samples
        end_sample = start_sample + slice_samples
        audio_slice = audio[start_sample:end_sample]

        output_file = os.path.join(output_dir, f'slice_{start_index + i:04d}.wav')
        sf.write(output_file, audio_slice, sample_rate)

    if len(audio) % slice_samples != 0:
        start_sample = num_slices * slice_samples
        audio_slice = audio[start_sample:]
        output_file = os.path.join(output_dir, f'slice_{start_index + num_slices:04d}.wav')
        sf.write(output_file, audio_slice, sample_rate)

if __name__ == "__main__":
    get_slices('rock_song_009.wav')