import subprocess
import os
import sys

def separate_audio(input_file,stems,output_folder="output"):
    try:
        # Create output directory if it doesn't exist
        os.makedirs(output_folder, exist_ok=True)
        
        input_file_name=input_file
        input_file_name= input_file_name.replace(".mp3","")
        # Use python -m to ensure the correct module is called
        command = [
            sys.executable, "-m", "spleeter", "separate", 
            "-o", output_folder+"\\"+input_file_name+stems,
            "-p", "spleeter:"+stems,
            "C:\\Users\\louis\\Desktop\\VScode\\Python\\Songs\\"+input_file, 
        ]
        
        print(f"Running command: {' '.join(command)}")
        subprocess.run(command, check=True)
        print(f"Separation complete! Files saved in {output_folder}/")
    except subprocess.CalledProcessError as e:
        print(f"Error running Spleeter: {e}")
    except Exception as e:
        print(f"Unexpected error: {e}")

# Example usage
separate_audio("TheSylversRememberTheRain.mp3" , "4stems")