[アップデート] Amazon BedrockでStable Diffusion 3.5 Largeが使えるようになりました！

import boto3
import json
import base64
import time
from datetime import datetime

bedrock = boto3.client(
    'bedrock-runtime',
    region_name='us-west-2'
)

request_body = {
    "prompt": "a beautiful flower",
    # "prompt": "A magical glowing flower in a mystical garden, iridescent petals, ethereal lighting, detailed botanical illustration style, photorealistic",
    "mode": "text-to-image",
    "aspect_ratio": "1:1",
    "output_format": "jpeg"
}

try:
    start_time = time.time()

    response = bedrock.invoke_model(
        modelId="stability.sd3-5-large-v1:0",
        # modelId="stability.sd3-large-v1:0",
        body=json.dumps(request_body),
        contentType="application/json",
        accept="application/json"
    )

    end_time = time.time()
    elapsed_time = round(end_time - start_time, 2)

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

    print("レスポンスの構造:")
    print(json.dumps(response_body, indent=2))

    if 'images' in response_body and response_body['images']:
        image_data = response_body['images'][0]

        timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
        filename = f"generated_flower_{timestamp}.jpg"

        with open(filename, "wb") as f:
            f.write(base64.b64decode(image_data))
        print(f"画像が正常に生成され、{filename}として保存されました。")
        print(f"処理時間: {elapsed_time}秒")
    else:
        print("画像データが見つかりませんでした。")

except Exception as e:
    print(f"エラーが発生しました: {str(e)}")
    import traceback
    print(traceback.format_exc())