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

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) # поиск сущностей
    paraphrase = pg.generate(srcText, srcEntities) # генерация парафраза
    paraEntities = ner.extract_entities(paraphrase) # поиск сущностей в парафразе
    return validator(srcText, paraphrase, srcEntities, paraEntities) # валидация

result = main(srcText)
print(f'ИСХОДНЫЙ ТЕКСТ: {srcText}')
print(f'СГЕНЕРИРОВАННЫЙ ТЕКСТ: {result}')