mkdir combined-layer
python3 -m venv env
source env/bin/activate #Windowsの場合はenv\Scripts\activate
pip install music21 requests more_itertools webcolors

mkdir python
cp -r env/lib/python3.12/site-packages/music21 python/
cp -r env/lib/python3.12/site-packages/requests python/
cp -r env/lib/python3.12/site-packages/charset_normalizer python/
cp -r env/lib/python3.12/site-packages/idna python/
cp -r env/lib/python3.12/site-packages/certifi python/
cp -r env/lib/python3.12/site-packages/more_itertools python/
cp -r env/lib/python3.12/site-packages/webcolors python/

zip -r music21-layer.zip python

import json
import boto3
import music21
from pathlib import Path
from datetime import datetime
import ast

s3 = boto3.client('s3')
bedrock_client = boto3.client(service_name='bedrock-runtime', region_name="us-east-1")

def generate_message(bedrock_runtime, model_id, messages, max_tokens, temp, top_p, top_k):
    body=json.dumps(
        {
            "anthropic_version": "bedrock-2023-05-31",
            "max_tokens": max_tokens,
            "messages": messages,
            "temperature": temp,
            "top_p": top_p,
            "top_k": top_k
        }  
    ) 
    response = bedrock_runtime.invoke_model(body=body, modelId=model_id)
    result = json.loads(response.get('body').read())
    input_tokens = result["usage"]["input_tokens"]
    output_tokens = result["usage"]["output_tokens"]
    output_list = result.get("content", [])

    print("Invocation details:")
    print(f"- The input length is {input_tokens} tokens.")
    print(f"- The output length is {output_tokens} tokens.")

    print(f"- The model returned {len(output_list)} response(s):")
    for output in output_list:
        print(output["text"])

    return result
    
def lambda_handler(event, context):
    # ユーザのプロンプト取得
    user_prompt = event.get('user_prompt')
    
    prompt = f"""\n\nHuman:
    あなたは優秀なゲームのBGM（背景音楽）メーカーです。ゲームの種類とシーンによって最適なABC記譜法（ノーテーション）を作ることが仕事です。
    <data></data>タグ内に記載された内容に基づいてメロディのノーテーションを出力します。
    <data>
    {user_prompt}
    </data>
    以下の<rule></rule>タグに従って出力してください。
    <rule>
    ・使用する楽器はmusic21.instrument.instrumentFromMidiProgramに基づいて楽器の番号を指定してください。
    ・楽器の番号と楽譜のノーテーションのみ出力してください。それ以外の文字は絶対に出力しないでください。
    ・ノーテーションについては、プレイヤーがゲームをプレイしている間、BGMが積極的にプレイヤーの注意を引きつけないよう注意してください。
    ・なるべく長めにノーテーションを書いてください。
    ・最低でも３つの楽器を使用してください。
    ・JSON文字列で出力してください。改行・空白は入れないでください。
    </rule>
    以下の</example></example>タグ内にノーテーションの出力の例になります。
    <example>
    81,[('C5', 0.36), ('E5', 0.36), ('G5', 0.36), ('F5', 0.36), ('D5', 0.36), ('C5', 0.36), (None, 0.36), ('G4', 0.36),('C5', 0.36), ('E5', 0.36), ('G5', 0.36), ('F5', 0.36), ('D5', 0.36), ('C5', 0.36), (None, 0.36), ('G4', 0.36),
                        ('A4', 0.36), ('C5', 0.36), ('D5', 0.36), ('B4', 0.36), ('G4', 0.36), ('A4', 0.36), (None, 0.36), ('F4', 0.36),
                        ('A4', 0.36), ('C5', 0.36), ('D5', 0.36), ('B4', 0.36), ('G4', 0.36), ('A4', 0.36), (None, 0.36), ('F4', 0.36)]

    88,[('C3', 0.71), (None, 0.71), ('G2', 0.71), (None, 0.71),
                  ('C3', 0.71), (None, 0.71), ('G2', 0.71), (None, 0.71),
                  ('F2', 0.71), (None, 0.71), ('C2', 0.71), (None, 0.71),
                  ('F2', 0.71), (None, 0.71), ('C2', 0.71), (None, 0.71)]

    116,[('C2', 0.36), (None, 0.36), ('C2', 0.36), (None, 0.36), ('C2', 0.36), (None, 0.36), ('C2', 0.36), (None, 0.36),
                  ('C2', 0.36), (None, 0.36), ('C2', 0.36), (None, 0.36), ('C2', 0.36), (None, 0.36), ('C2', 0.36), (None, 0.36),
                  ('C2', 0.71), (None, 0.71), ('C2', 0.71), (None, 0.71),
                  ('C2', 0.71), (None, 0.71), ('C2', 0.71), (None, 0.71)]

    115,[(None, 0.71), ('D4', 0.18), (None, 0.18), ('D4', 0.18), (None, 0.18),
                   (None, 0.71), ('D4', 0.18), (None, 0.18), ('D4', 0.18), (None, 0.18),
                   (None, 0.71), ('D4', 0.18), (None, 0.18), ('D4', 0.18), (None, 0.18),
                   (None, 0.71), ('D4', 0.18), (None, 0.18), ('D4', 0.18), (None, 0.18),
                   (None, 0.71), ('D4', 0.18), (None, 0.18), ('D4', 0.18), (None, 0.18),
                   (None, 0.71), ('D4', 0.18), (None, 0.18), ('D4', 0.18), (None, 0.18)]
    </example>

    Assistant:
    """
    
    ### Defining the messages and invoking the model
    messages=[{ "role":'user', "content":[{'type':'text','text': prompt}]}]
    result = generate_message(
        bedrock_client, model_id = "anthropic.claude-3-sonnet-20240229-v1:0",
        messages=messages,
        max_tokens=4096,
        temp=1,
        top_p=1,
        top_k=100)
        
    response_text = ""
    for chunk in result.get("content", []):
        response_text += chunk.get("text", "")
    
    response_json = ast.literal_eval(response_text)
    
    bgm_score = music21.stream.Score()

    # 数字キーと対応する値を辞書に格納
    numeric_keys = {}
    for key, value in response_json.items():
       if key.isdigit():
           numeric_keys[int(key)] = value

    # 楽器と音価データを抽出
    instruments = []
    for key, value in numeric_keys.items():
        key
        value
        instrument = music21.instrument.instrumentFromMidiProgram(key)
        instruments.append((value, 0.0, instrument))
    
    for notes, offset, instrument in instruments:
        instrument_part = music21.stream.Part()
        instrument_part.insert(0, instrument)
        for pitch, duration in notes:
            if pitch is not None:
                note = music21.note.Note(pitch, quarterLength=duration)
                instrument_part.append(note)
            else:
                rest = music21.note.Rest(quarterLength=duration)
                instrument_part.append(rest)
        bgm_score.insert(offset, instrument_part)
    
    # 音価データをMIDIファイルに書き込む
    midi_file = bgm_score.write('midi')
    midi_data = open(midi_file, 'rb').read()
    
    # タイムスタンプ付きのMIDIファイル名を定義する
    current_datetime = datetime.now().strftime('%Y%m%d-%H%M%S')
    midi_file_name = f'bgm_{current_datetime}.mid'
    bucket_name = 'artifacts-bgm-generator'
    
    s3.put_object(
    Bucket=bucket_name,
    Key=midi_file_name,
    Body=midi_data
    )
    
    return {
        'statusCode': 200,
        'body': json.dumps(f'BGM MIDI file uploaded to S3://{bucket_name}/{midi_file_name}')
    }