Building SunSwitch: AI-Powered DG-to-Solar Switch Advisor Using Amazon Bedrock & AWS Serverless

import json
import boto3

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

def extract_logbook_data(bucket_name: str, object_key: str):
    # Fetch image from ephemeral S3 storage
    s3_obj = s3.get_object(Bucket=bucket_name, Key=object_key)
    image_bytes = s3_obj["Body"].read()

    request_payload = {
        "messages": [
            {
                "role": "user",
                "content": [
                    {"image": {"format": "jpeg", "source": {"bytes": image_bytes}}},
                    {"text": "Extract total diesel litres, operating hours, and duration in months from this logbook. Return valid JSON only with keys: total_litres, total_hours, period_months, confidence."}
                ]
            }
        ],
        "inferenceConfig": {"maxTokens": 512, "temperature": 0.1}
    }

    response = bedrock.invoke_model(
        modelId="us.amazon.nova-lite-v1:0",
        body=json.dumps(request_payload)
    )

    result = json.loads(response["body"].read())
    return json.loads(result["output"]["message"]["content"][0]["text"])