add validator

main.py

@@ -1,26 +1,31 @@
 # Точка входа программы
 
-from NER import NER
-from paraGenerator import ParaphraseGenerator
+from modules.NER import NER
+from modules.paraGenerator import ParaphraseGenerator
+#from modules.validator import validator
 
-## Запуск рабочих инструментов
 ner = NER()
 pg = ParaphraseGenerator()
-###
 
 srcText = 'Добрый день, я, Сидоров Иван Иванович. Прошу перевести сто тысяч рублей Якову Петру Игнатьевичу в Москву.'
 
-entities1 = ner.extract_entities(srcText)
-print(entities1)
+def main(srcText):
+    # поиск сущностей
+    srcEntities = ner.extract_entities(srcText)
+    print(srcEntities)
 
-#paraphrase1 = pg.generate(srcText, entities1)
-#print(paraphrase1)
+    # генерация парафраза
+    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)
 
-#entities2 = ner.extract_entities(paraphrase1)
-#print(entities2)
 
-gg = ()
-gg.append(1)
-gg.append(2)
-gg.append(1)
-print(gg)