AI Music Looper

What It Does

Installation

Usage

python ai_music_looper.py --input track.mp3 --preview

Suggested loop points:
Start: 0.00s, End: 1.00s
Start: 1.00s, End: 2.00s
...
Previewing loops for file: track.mp3
Loop from 0.00s to 1.00s
Loop from 1.00s to 2.00s
...

Source Code

import argparse
import librosa
import numpy as np

def analyze_music(file_path):
    """
    Analyze the music file to find repeating patterns and suggest loop points.

    Args:
        file_path (str): Path to the music file.

    Returns:
        list of tuple: List of tuples containing start and end timestamps of loop points.
    """
    try:
        # Load the audio file
        y, sr = librosa.load(file_path, sr=None)

        # Compute the onset envelope
        onset_env = librosa.onset.onset_strength(y=y, sr=sr)

        # Compute the tempogram
        tempo, beats = librosa.beat.beat_track(onset_envelope=onset_env, sr=sr)

        # Convert beat frames to timestamps
        beat_times = librosa.frames_to_time(beats, sr=sr)

        # Find repeating patterns (naive approach: consecutive beats)
        loop_points = [(beat_times[i], beat_times[i + 1]) for i in range(len(beat_times) - 1)]

        return loop_points

    except Exception as e:
        raise RuntimeError(f"Error analyzing music file: {e}")

def preview_loops(file_path, loop_points):
    """
    Preview the suggested loop points by printing them.

    Args:
        file_path (str): Path to the music file.
        loop_points (list of tuple): List of start and end timestamps for loops.
    """
    print(f"Previewing loops for file: {file_path}")
    for start, end in loop_points:
        print(f"Loop from {start:.2f}s to {end:.2f}s")

def main():
    parser = argparse.ArgumentParser(description="AI Music Looper: Analyze music files for seamless loop points.")
    parser.add_argument("--input", required=True, help="Path to the input music file (e.g., WAV, MP3).")
    parser.add_argument("--preview", action="store_true", help="Preview the suggested loop points.")

    args = parser.parse_args()

    try:
        # Analyze the music file
        loop_points = analyze_music(args.input)

        # Print the loop points
        print("Suggested loop points:")
        for start, end in loop_points:
            print(f"Start: {start:.2f}s, End: {end:.2f}s")

        # Preview the loop points if requested
        if args.preview:
            preview_loops(args.input, loop_points)

    except RuntimeError as e:
        print(e)

if __name__ == "__main__":
    main()