main_r.py

import librosa
import IPython.display as ipd  # For playing audio

# Load the audio file
audio_file = '/content/drive/MyDrive/autochord-0.1.4/moonarch/SS_2.mp3'
y, sr = librosa.load(audio_file, sr=None)  # sr=None to use the native sample rate

choice = int(input("Enter choice from ['vocal_remove', 'basic', 'advance'] as 0, 1, or 2: "))

if choice == 0:
    from moonarch_vocal_remover import VocalRemover

    # Process the audio for vocal removal
    splitter = VocalRemover(audio_file)
    splitter.run()

    play_aud = os.path.splitext(audio_file)[0]  # without extension

    print("Vocal")
    ipd.display(ipd.Audio(f'{play_aud}/vocals.wav'))
    print("Music")
    ipd.display(ipd.Audio(f'{play_aud}/accompaniment.wav'))

elif choice == 1:
    from moonarch_basic import BasicSplitter
    from moonarch_vocal_remover import VocalRemover

    # Process the audio for basic split
    splitter = BasicSplitter(audio_file)
    splitter.run()

    # Process the audio for vocal removal
    music_sep = VocalRemover(audio_file)
    music_sep.run()

    play_aud = os.path.splitext(audio_file)[0]  # without extension

    print("Vocal")
    ipd.display(ipd.Audio(f'{play_aud}/vocals.wav'))
    print("Music")
    ipd.display(ipd.Audio(f'{play_aud}/accompaniment.wav'))
    print("Bass")
    ipd.display(ipd.Audio(f'{play_aud}/bass.wav'))
    print("Drums")
    ipd.display(ipd.Audio(f'{play_aud}/drums.wav'))

elif choice == 2:
    from moonarch_advance import AdvanceSplitter

    # Process the audio for advanced split
    splitter = AdvanceSplitter(audio_file)
    splitter.run()

    play_aud = os.path.splitext(audio_file)[0]  # without extension

    print("Vocal")
    ipd.display(ipd.Audio(f'{play_aud}/vocals.wav'))
    print("Strings")
    ipd.display(ipd.Audio(f'{play_aud}/other.wav'))
    print("Bass")
    ipd.display(ipd.Audio(f'{play_aud}/bass.wav'))
    print("Drums")
    ipd.display(ipd.Audio(f'{play_aud}/drums.wav'))
    print("Piano")
    ipd.display(ipd.Audio(f'{play_aud}/piano.wav'))

print("Audio processed successfully.")