Bedrock×Rekognitionで、画像認識して関西弁でツッコむ「辛口AIソムリエ」を作ってみた

# 命令
あなたは辛口な関西人として、AIソムリエの役割を完璧に演じてください。
# あなたのキャラクター設定 (ペルソナ)
- **口調:** 常に横柄でドSな関西弁。一人称は「俺」、相手のことは「お前」と呼ぶ。
- **性格:** 短気でせっかち。回りくどいことは大嫌い。面白いかどうかを常に考えている。
- **口癖:** 「なんでやねん！」「アホか！」「知るか！」「しゃあないな」などを自然な会話の流れで使うこと。
- **行動:**
1. まず、目の前にある食べ物（入力情報）に対して、強烈な一言（イジりやツッコミ）をかます。例：「なんやこれ、茶色いな！」「こんなもん夜中に食うとんのか、アホやろ！」など。
2. 次に「まぁしゃあないから、この俺が特別に合う飲み物を選んだるわ」というような、恩着せがましい前置きを入れる。
3. 最後に、最高のドリンクを3つ(アルコール飲料とソフトドリンクをバランスよく)提案する。提案理由は的確で分かりやすいが、言い方は偉そうにすること。
# 入力情報
- Rekognitionが画像から検出した食べ物の特徴: {', '.join(labels)}
# 出力タスク
上記のキャラクター設定を忠実に守り、入力情報に対して最高のドリンク提案を生成してください。

import json
import urllib.request
import boto3

# 各サービスクライアントを準備
# (Bedrockはバージニア北部リージョン 'us-east-1' が推奨です)
bedrock_runtime = boto3.client('bedrock-runtime', region_name='us-east-1')
rekognition = boto3.client('rekognition')

def handler(event, context):
    try:
        # 1. 画像URLを受け取る
        body = json.loads(event.get('body', '{}'))
        image_url = body.get('imageUrl')

        # 2. Rekognitionで画像からキーワードを検出する
        with urllib.request.urlopen(image_url) as response:
            image_bytes = response.read()
        
        response_rekognition = rekognition.detect_labels(
            Image={'Bytes': image_bytes},
            MaxLabels=10, MinConfidence=80
        )
        labels = [label['Name'] for label in response_rekognition['Labels']]

        # 3. 上記で解説した「プロンプト」を作成する
        #    labelsには ['Pizza', 'Cheese', 'Food'] などが入ります
        prompt = f"""
# 命令
あなたは辛口の関西人として...(中略)...

# 入力情報
- Rekognitionが画像から検出した食べ物の特徴: {', '.join(labels)}

# 出力タスク
...(後略)...
"""
        # 4. Bedrockにプロンプトを渡し、応答を生成してもらう
        model_id = 'anthropic.claude-3-sonnet-20240229-v1:0'
        request_body = {
            "anthropic_version": "bedrock-2023-05-31",
            "max_tokens": 1000,
            "messages": [{"role": "user", "content": [{"type": "text", "text": prompt}]}]
        }
        response_bedrock = bedrock_runtime.invoke_model(
            body=json.dumps(request_body), modelId=model_id,
        )
        response_body = json.loads(response_bedrock.get('body').read())
        suggestion_text = response_body['content'][0]['text']

        # 5. 生成された文章をユーザーに返す
        return {
            'statusCode': 200,
            'headers': { 'Access-Control-Allow-Origin': '*', 'Content-Type': 'application/json' },
            'body': json.dumps({'suggestion': suggestion_text})
        }
    
    except Exception as e:
        print(f"Error: {e}")
        return {'statusCode': 500, 'body': json.dumps({'error': str(e)})}