maolan-generate 0.0.11

Generate music using Maolan and HeartMuLa
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

//! Text-to-MIDI generation for `maolan-generate`.
//!
//! This module implements the AMT (Anticipatory Music Transformer) arrival-time
//! tokenization used by MIDI-LLM and provides two generators:
//!
//! - a lightweight, deterministic prompt interpreter (`generator`)
//! - a real MIDI-LLM Llama 3.2 1B model loader and sampler (`midi_llm`)
//!
//! Both generators emit the same AMT token format, so they share the MIDI
//! writer in `midi`.

pub mod amt;
pub mod generator;
pub mod hf_tokenizer;
pub mod midi;
pub mod midi_llm;

pub use amt::{AmtEvent, events_to_tokens, tokens_to_events};
pub use generator::{TextToMidiConfig, generate_events, generate_tokens};
pub use midi::{TimeSignature, write_midi};
pub use midi_llm::{MidiLlmConfig, generate_midi_file as generate_midi_file_with_llm, load_model};

use anyhow::{Context, Result};
use std::path::Path;

/// Generate a MIDI file from `prompt` and write it to `output_path`.
///
/// `bpm`, `key_scale`, and `time_signature` may be extracted from the prompt or
/// supplied explicitly. `length_seconds` caps the generated sequence.
pub fn generate_midi_file<P: AsRef<Path>>(
    output_path: P,
    prompt: &str,
    bpm: Option<f32>,
    key_scale: Option<&str>,
    time_signature: Option<&str>,
    length_seconds: f32,
    seed: u64,
) -> Result<()> {
    let config = TextToMidiConfig {
        prompt,
        bpm,
        key_scale,
        time_signature,
        length_seconds,
        seed,
    };
    let events = generate_events(&config);
    let ts = time_signature.and_then(TimeSignature::parse);
    let effective_bpm = bpm.unwrap_or_else(|| {
        // Re-run lightweight BPM detection when no explicit value was given.
        let words: Vec<String> = prompt
            .to_ascii_lowercase()
            .split(|c: char| !c.is_alphanumeric())
            .map(str::to_owned)
            .filter(|w| !w.is_empty())
            .collect();
        generator::detect_bpm(&words, None)
    });
    write_midi(output_path, &events, effective_bpm, ts)
        .with_context(|| "failed to write generated MIDI file")
}