21 lines
933 B
Python
21 lines
933 B
Python
# Точка входа программы
|
||
|
||
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}')
|