def main():
    if len(sys.argv) != 2:
        print("Usage: python3 helper-infra-cost-estimation.py terraform-config-rds.json")
        sys.exit(1)

    config_file = sys.argv[1]
    try:
        with open(config_file, 'r') as f:
            tf_json = json.load(f)
    except Exception as ex:
        print(f"Could not read config file: {ex}")
        sys.exit(2)

    input_text = json.dumps(tf_json, indent=2)
    bedrock_agent = boto3.client('bedrock-agent-runtime', region_name=REGION)

    print("\n=== Input ===")
    print(input_text)

    try:
        response = bedrock_agent.invoke_agent(
            agentId=AGENT_ID,
            agentAliasId=AGENT_ALIAS_ID,
            sessionId='infra-cost-estimation-session',
            inputText='Create summary estimate cost'+input_text
        )
        event_stream = response["completion"]
        # When iterated the EventStream will yield events.

        print("\n=== Output form Bedrock agent ===")
        for event in event_stream:

            # chunk contains a part of an agent response
            if 'chunk' in event:
                chunk = event['chunk']
                if 'bytes' in chunk:
                    text = chunk['bytes'].decode('utf-8')
                    print(f"Chunk: {text}")
                else:
                    print("Chunk doesn't contain 'bytes'")

    except Exception as ex:
        print(f"Agent invocation failed: {ex}")
        sys.exit(3)

if __name__ == "__main__":
    main()