import plotwave

plotwave.plot(
    [
        plotwave.audio_file("examples/plotwave_C_piano.wav", name="Piano", color="#2563eb"),
        plotwave.audio_file(
            "examples/plotwave_C_VHS.wav",
            name="VHS-style synth",
            color="#ea580c",
            line={"dash": "dot"},
        ),
        plotwave.segments(
            [
                {"start": 0.0, "end": 2.09, "label": "C"},
                {"start": 2.09, "end": 4.18, "label": "Asus2"},
                {"start": 4.18, "end": 6.27, "label": "Esus2"},
                {"start": 6.27, "end": 8.3, "label": "B"},
            ],
            name="Ground Truth",
            lane="top",
            color_map={
                "C": "#0f62fe",
                "Asus2": "#367af7",
                "Esus2": "#82adfc",
                "B": "#bcd2fb",
            },
        ),
        plotwave.segments(
            [
                {"start": 0.0, "end": 2.09, "label": "C"},
                {"start": 2.09, "end": 4.18, "label": "D5"},
                {"start": 4.18, "end": 6.27, "label": "Esus2"},
                {"start": 6.27, "end": 8.3, "label": "B"},
            ],
            name="Prediction",
            lane="bottom",
            color_map={
                "C": "#0f62fe",
                "D5": "#ee9f0d",
                "Esus2": "#82adfc",
                "B": "#bcd2fb",
            },
        ),
    ],
    layout={"title": {"text": "Style transfer: piano to VHS-style synth"}},
)