Amazon Comprehend Medical でアーティストの問診票作ってみた。

以下のコマンドを入力してファイルを作成します。

nano doctor_jpop.py

開いたエディタに、以下のコードを丸ごとコピー＆ペーストしてください。

import boto3
import time

# ==========================================
# 診断対象のプレイリスト
# ==========================================
PLAYLIST = [
    {
        "Artist": "失恋太郎",
        "Song": "眠れない夜",
        "Lyrics": "胸が張り裂けそうで眠れない夜。食事が喉を通らない。"
    },
    {
        "Artist": "炎のロッカー",
        "Song": "バーニング・ラブ",
        "Lyrics": "燃えるような恋心。叫びすぎて声が枯れた。息ができない。"
    },
    {
        "Artist": "純愛ちゃん",
        "Song": "1000年の愛",
        "Lyrics": "1000年先も愛を誓う。君の光で目が眩む。体が砕けても守る。"
    },
    {
        "Artist": "健康キッズ",
        "Song": "お野菜のうた",
        "Lyrics": "お日様を浴びて、トマトとキャベツを食べよう。元気いっぱい。"
    },
    {
        "Artist": "激しいメタル",
        "Song": "ブラッディ・レイン",
        "Lyrics": "頭が割れるように痛い。血が騒ぐ。毒のようなキス。意識が遠のく。"
    }
]

# 設定エリア
TARGET_REGION = 'us-east-1' # Comprehend Medical用

def main():
    # クライアント初期化
    translate = boto3.client('translate', region_name=TARGET_REGION)
    medical = boto3.client('comprehendmedical', region_name=TARGET_REGION)

    print(f"--- J-POP 健康診断を開始します（全 {len(PLAYLIST)} 曲） ---\n")

    # 結果格納用リスト
    ranking_data = []

    for track in PLAYLIST:
        artist = track['Artist']
        song = track['Song']
        lyrics = track['Lyrics']

        print(f"Now Checking... [{artist} / {song}]")

        # 1. 翻訳 (JA -> EN)
        try:
            trans_resp = translate.translate_text(
                Text=lyrics, SourceLanguageCode='ja', TargetLanguageCode='en'
            )
            en_text = trans_resp['TranslatedText']
        except Exception as e:
            print(f" 翻訳エラー: {e}")
            continue

        # 2. 診断 (Comprehend Medical)
        try:
            med_resp = medical.detect_entities_v2(Text=en_text)
            entities = med_resp['Entities']
        except Exception as e:
            print(f" 診断エラー: {e}")
            continue

        # 3. 集計
        symptom_count = len(entities)

        # どんな症状が見つかったかテキストでまとめる
        symptoms_list = [e['Text'] for e in entities]
        symptoms_str = ", ".join(symptoms_list) if symptoms_list else "なし（健康）"

        ranking_data.append({
            "Artist": artist,
            "Song": song,
            "Score": symptom_count,
            "Details": symptoms_str
        })

        # APIレート制限考慮で少し待機
        time.sleep(0.5)

    # ==========================================
    # 結果発表（ランキング表示）
    # ==========================================
    ranking_data.sort(key=lambda x: x['Score'], reverse=True)

    print("\n" + "="*80)
    print(f"{'RANK':<5} | {'ARTIST':<15} | {'SCORE (危険度)':<15} | {'DIAGNOSIS (主な症状)'}")
    print("="*80)

    for i, data in enumerate(ranking_data, 1):
        score = data['Score']
        if score == 0:
            judge = "健康優良児"
        elif score < 3:
            judge = f"要経過観察 ({score})"
        else:
            judge = f"即入院 ({score})"

        print(f"{i:<5} | {data['Artist']:<15} | {judge:<15} | {data['Details']}")

    print("="*80)

if __name__ == "__main__":
    main()

以下のコマンドでスクリプトを実行します。

python3 doctor_jpop.py