DJ Stem Separation Workflow

import requests
import os
import time
from pathlib import Path

API_KEY = "sk_live_your_api_key"
INPUT_DIR = "./tracks"          # Your source tracks
OUTPUT_DIR = "./stems"          # Where stems will be saved
BASE_URL = "https://stemsplit.io/api/v1"

# What to extract (choose one)
# OPTIONS: "VOCALS", "INSTRUMENTAL", "ALL_STEMS"
OUTPUT_TYPE = "ALL_STEMS"

def upload_and_process(file_path):
    """Upload a track and get stems"""
    filename = os.path.basename(file_path)
    print(f"\n🎵 Processing: {filename}")
    
    # Step 1: Get presigned upload URL
    upload_response = requests.post(
        f"{BASE_URL}/upload",
        headers={"Authorization": f"Bearer {API_KEY}"},
        json={
            "filename": filename,
            "contentType": "audio/mpeg"
        }
    )
    upload_data = upload_response.json()
    
    # Step 2: Upload the file
    with open(file_path, "rb") as f:
        requests.put(upload_data["uploadUrl"], data=f)
    
    # Step 3: Create job
    job_response = requests.post(
        f"{BASE_URL}/jobs",
        headers={"Authorization": f"Bearer {API_KEY}"},
        json={
            "uploadKey": upload_data["key"],
            "outputType": OUTPUT_TYPE,
            "quality": "BEST",
            "outputFormat": "MP3"
        }
    )
    job = job_response.json()
    job_id = job["id"]
    
    # Step 4: Wait for completion
    while True:
        status = requests.get(
            f"{BASE_URL}/jobs/{job_id}",
            headers={"Authorization": f"Bearer {API_KEY}"}
        ).json()
        
        if status["status"] == "COMPLETED":
            break
        elif status["status"] == "FAILED":
            print(f"  ❌ Failed: {status.get('error', 'Unknown error')}")
            return
        
        print(f"  ⏳ Status: {status['status']}...")
        time.sleep(10)
    
    # Step 5: Download stems
    track_name = Path(filename).stem
    track_dir = os.path.join(OUTPUT_DIR, track_name)
    os.makedirs(track_dir, exist_ok=True)
    
    for stem_name, stem_data in status["outputs"].items():
        output_path = os.path.join(track_dir, f"{stem_name}.mp3")
        audio = requests.get(stem_data["url"]).content
        with open(output_path, "wb") as f:
            f.write(audio)
        print(f"  ✓ Saved: {stem_name}.mp3")
    
    print(f"  ✅ Complete! Stems saved to: {track_dir}")

# Process all tracks in input directory
def main():
    os.makedirs(OUTPUT_DIR, exist_ok=True)
    
    audio_extensions = {".mp3", ".wav", ".flac", ".m4a", ".aac"}
    tracks = [f for f in os.listdir(INPUT_DIR) 
              if Path(f).suffix.lower() in audio_extensions]
    
    print(f"Found {len(tracks)} tracks to process")
    
    for track in tracks:
        upload_and_process(os.path.join(INPUT_DIR, track))
        time.sleep(2)  # Small delay between tracks
    
    print(f"\n🎉 All done! Check {OUTPUT_DIR} for your stems")

if __name__ == "__main__":
    main()