# Точка входа программы

from modules.NER import NER
from modules.paraGenerator import ParaphraseGenerator
#from modules.validator import validator

ner = NER()
pg = ParaphraseGenerator()

srcText = 'Добрый день, я, Сидоров Иван Иванович. Прошу перевести сто тысяч рублей Якову Петру Игнатьевичу в Москву.'

def main(srcText):
    # поиск сущностей
    srcEntities = ner.extract_entities(srcText)
    print(srcEntities)

    # генерация парафраза
    paraphrase = pg.generate(srcText, srcEntities)
    print(paraphrase)

    # поиск сущностей в парафразе
    paraEntities = ner.extract_entities(paraphrase)
    print(paraEntities)

    # Валидация
 #   return validator(srcText, paraphrase, srcEntities, paraEntities)

result = main(srcText)
print(result)