from transformers import BartTokenizer
from idiomify.builders import SourcesBuilder

BATCH = [
    ("I could die at any moment", "I could kick the bucket at any moment"),
    ("Speak plainly", "Don't beat around the bush")
]


def main():
    tokenizer = BartTokenizer.from_pretrained("facebook/bart-large")
    builder = SourcesBuilder(tokenizer)
    src = builder(BATCH)
    print(src)


if __name__ == '__main__':
    main()