32 lines
928 B
Python
32 lines
928 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)
|
|
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)
|
|
|
|
|