.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/musiclang_api/07_continue_song.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_auto_examples_musiclang_api_07_continue_song.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_musiclang_api_07_continue_song.py:


Extend an existing sample or song
======================================================

In this example :
- We load an existing song
- We ask musiclang to continue it for 4 bars

.. GENERATED FROM PYTHON SOURCE LINES 10-47

.. code-block:: Python


    import os
    from maidi import MidiScore, instrument, midi_library
    from maidi.integrations.api import MusicLangAPI
    from maidi import chords_symbols as cs

    # Assuming API_URL and MUSICLANG_API_KEY are set in the environment
    MUSICLANG_API_KEY = os.getenv("MUSICLANG_API_KEY")

    nb_bars_extension = 4

    # Load a score with two tracks and 4 bars (drum and bass)
    score = MidiScore.from_midi(midi_library.get_midi_file('drum_and_bass'))
    print("Starting number of bars : ", score.nb_bars)
    # Control the chord progression, same as before (Optional)
    chords = [
                (cs.I, cs.C, cs.major, cs._root_position),
               (cs.VI, cs.C, cs.major, ''), # Triad in root position can also be written as empty string ''
               (cs.II, cs.C, cs.major, cs._6),
               (cs.V, cs.C, cs.major, cs._7),
    ]

    # Add some tags for the new bars (Optional)
    tags = [[['CONTROL_MIN_POLYPHONY__1', 'CONTROL_MAX_POLYPHONY__1', 'CONTROL_DENSITY__LOW'] for bar_index in range(nb_bars_extension)]
            for track_index in range(score.nb_tracks)
            ]

    # Call the musiclang API to predict the score
    api = MusicLangAPI(api_key=MUSICLANG_API_KEY, verbose=True)
    predicted_score = api.extend(score,
                                 nb_bars_extension,  #  How many bars to add to the current score
                                 chords=chords,
                                 tags=tags,
                                 nb_added_bars_step=2,  # Choose how many bar generate per steps (if none it is calculated automatically)
                                 polling_interval=3
    )
    print("Final number of bars : ", predicted_score.nb_bars)
    predicted_score.write("predicted_score.mid")

.. _sphx_glr_download_auto_examples_musiclang_api_07_continue_song.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: 07_continue_song.ipynb <07_continue_song.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: 07_continue_song.py <07_continue_song.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: 07_continue_song.zip <07_continue_song.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_