from transformers import pipeline
question_answerer = pipeline("question-answering", model='bert-large-uncased-whole-word-masking-finetuned-squad')

context = r"""
cube1 is visible.
cube1 orientation is upright.
cube1 position is (200.5, 11.3).
cube2 is not visible.
cube2 orientation is upright.
cube2 position is (150.5, 34.5).
cube3 is visible.
cube3 orientation is sideways.
cube3 position is (-50.5, 138.7).
"""

print(context)

while (True):
    query = input('Query? ')
    if len(query) == 0: continue
    result = question_answerer(question=query, context=context)
    score = round(result['score'], 4)
    if score > 0.3:
        print(f"Answer: '{result['answer']}', score: {round(result['score'], 4)}, start: {result['start']}, end: {result['end']}")
    else:
        print(f"I don't know.  score={score}")
    print()