Amazon PollyとAmazon Bedrockを使ってセルフAudibleやってみた

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "bedrock:InvokeModel",
                "bedrock:ListFoundationModels"
            ],
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "polly:SynthesizeSpeech",
                "polly:DescribeVoices"
            ],
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "s3:PutObject",
                "s3:GetObject",
                "s3:ListBucket"
            ],
            "Resource": [
                "arn:aws:s3:::self-audible-bucket/*",
                "arn:aws:s3:::self-audible-bucket"
            ]
        }
    ]
}

import boto3
import json
import os

S3_BUCKET_NAME = os.environ.get('S3_BUCKET_NAME')
BEDROCK_MODEL_ID = os.environ.get('BEDROCK_MODEL_ID')
POLLY_VOICE_ID = os.environ.get('POLLY_VOICE_ID', 'Takumi')

def lambda_handler(event, context):
    # テーマとジャンルを取得
    theme = event.get('theme', '未来')
    genre = event.get('genre', 'SF')

    story = generate_short_story(theme, genre)

    title, content = extract_title_and_content(story)

    # 音声変換
    audio_url = text_to_speech(content, title)

    return {
        'statusCode': 200,
        'body': json.dumps({
            'title': title,
            'story': content,
            'audio_url': audio_url
        }, ensure_ascii=False)
    }

def generate_short_story(theme, genre):
    bedrock = boto3.client('bedrock-runtime', region_name='ap-northeast-1')

    # システムプロンプト
    system_prompt = """あなたは優れたショートショート作家です。
条件:
- 800文字程度の短いショートショートにしてください
- 起承転結のある完結した物語にしてください
- 読者の心に残るような印象的な結末を用意してください
- 日本語で書いてください

必ず以下の形式で出力してください:
タイトル: [あなたが考えたタイトル]

[本文]

必ず「タイトル: 」で始まる形式を守ってください。
タイトルは短く、ファイル名に使えるようにしてください（特殊文字は避けてください）。"""

    # メッセージ
    user_message = f"""以下の条件で短編小説を書いてください。

テーマ: {theme}
ジャンル: {genre}"""

    # リクエストの作成
    request = {
        "anthropic_version": "bedrock-2023-05-31",
        "max_tokens": 4000,
        "temperature": 0.7,
        "system": system_prompt,
        "messages": [
            {
                "role": "user",
                "content": user_message
            }
        ]
    }

    response = bedrock.invoke_model(
        modelId=BEDROCK_MODEL_ID,
        body=json.dumps(request)
    )

    response_body = json.loads(response['body'].read())

    # ストーリー抽出
    story = response_body['content'][0]['text']

    return story

def extract_title_and_content(story):
    lines = story.strip().split('\n')
    title = ""
    content = ""

    # 最初の行が「タイトル:」で始まることを想定
    if lines and lines[0].startswith("タイトル:"):
        title = lines[0].replace("タイトル:", "").strip()
        content = "\n".join(lines[1:]).strip()
    else:
        title = "無題のショートショート"
        content = "\n".join(lines).strip()

    return title, content

def text_to_speech(text, title):
    polly = boto3.client('polly')
    s3 = boto3.client('s3')

    # ファイル名の作成
    file_name = f"{title}.mp3"

    # 音声合成リクエスト
    response = polly.synthesize_speech(
        Text=text,
        OutputFormat="mp3",
        VoiceId=POLLY_VOICE_ID,
        Engine="neural"  # ニューラル音声
    )

    # 一時ファイルに保存
    temp_file = "/tmp/story.mp3"

    try:
        # 音声データをファイルに保存
        with open(temp_file, "wb") as file:
            file.write(response["AudioStream"].read())

        # S3にアップロード
        s3.upload_file(temp_file, S3_BUCKET_NAME, file_name)

        # S3のURLを生成
        s3_url = f"https://{S3_BUCKET_NAME}.s3.amazonaws.com/{file_name}"

        return s3_url

    except Exception as e:
        print(f"音声変換エラー: {str(e)}")
        raise e
    finally:
        # 一時ファイルを削除
        if os.path.exists(temp_file):
            os.remove(temp_file)

"あなたは優れたショートショート作家です。
条件:
- 800文字程度の短いショートショートにしてください
- 起承転結のある完結した物語にしてください
- 読者の心に残るような印象的な結末を用意してください
- 日本語で書いてください

必ず以下の形式で出力してください:
タイトル: [あなたが考えたタイトル]

[本文]

必ず「タイトル: 」で始まる形式を守ってください。
タイトルは短く、ファイル名に使えるようにしてください（特殊文字は避けてください）。"

{
  "theme": "時間旅行",
  "genre": "SF"
}

{
  "statusCode": 200,
  "body": {
    "title": "時の扉", 
    "story": "「おじいちゃん、これは何？」孫の真也が、古い木箱に入った奇妙な懐中時計を見つけた。\n\n「それはな、時間を旅する時計だよ」と私は答えた。真也は目を輝かせる。子供の好奇心をくすぐる話だ。\n\n「嘘だー」と真也は笑う。\n\n「本当だよ。でも一つだけ決まりがある。過去は変えられない。どんなに辛くても、ただ見るだけなんだ」\n\n真也は半信半疑で時計を手に取った。「使い方は？」\n\n「針を回して日付を決め、ボタンを押すだけさ」\n\nその夜、私は真也が寝静まった後、こっそりと時計を取り出した。針を回し、30年前の日付に合わせる。ボタンを押すと、周囲が霞み、光の渦に包まれた。\n\n目が覚めると、そこは病院だった。ベッドに横たわる妻。そして30歳若い自分が泣きながら手を握っている。妻の最期の日だ。\n\n「あなた、約束して」かすれた声で妻が言う。「真也をしっかり育ててね」\n\n若い私は泣きながら頷いた。\n\n私は見えない存在として、その場面をただ見つめることしかできない。変えられないと知りながら、もう一度妻の声を聞きたかった。\n\n光の渦が再び現れ、私は現在に戻った。枕元には真也が立っていた。\n\n「おじいちゃん、泣いてるの？」\n\n「少し昔を思い出してね」と私は微笑んだ。\n\n「僕も試してみたよ。おばあちゃんに会えた」真也の目には涙が光っていた。「おばあちゃん、僕のこと知ってた。『真也、おじいちゃんを頼むね』って言ってた」\n\n驚いて顔を上げると、真也は続けた。「時間を変えられなくても、見守ることはできるんだね」\n\n私たちは黙って抱き合った。時計は静かに刻み続ける。過去は変えられなくても、記憶は未来へと流れ続けるのだから。", 
    "audio_url": "https://self-audible-bucket.s3.amazonaws.com/時の扉.mp3"
  }
}