fix errors

modules/validator.py

@@ -1,7 +1,7 @@
 # 3 модуль - Валидация и восстановление сущностей
 
-from NER import NER
-from paraGenerator import ParaphraseGenerator
+from modules.NER import NER
+from modules.paraGenerator import ParaphraseGenerator
 from pymorphy3 import MorphAnalyzer
 
 ner = NER()
@@ -45,9 +45,9 @@ def compare_entities(original, generated):
             lost.add(o)
     return False, lost
 
-def validator(srcText, srcEntities, paraEntities):
+def validator(srcText, paraphrase, srcEntities, paraEntities):
     """
-    Использование: validator(<исходный текст>, <сущности исходного текста>, <сущности перефразированного текста>)
+    Использование: validator(<исходный текст>, <парафразированный текст> <сущности исходного текста>, <сущности перефразированного текста>)
 
     Возвращает:
     - Исходный текст если сущности сохранены
@@ -56,10 +56,14 @@ def validator(srcText, srcEntities, paraEntities):
     """
     ce = compare_entities(srcEntities, paraEntities)
     if ce[0]:
-        return srcText # если всё нормально, возвращаем текст в неизменном виде
+        return paraphrase # если всё нормально, возвращаем текст в неизменном виде
     
-    # даём 3 попытки на восстановление
-    for _ in range(3):
+    print(f'Произошла потеря сущностей!')
+    print(f'Исходные сущности: {srcEntities}')
+    print(f'Сгенерируемые сущности: {paraEntities}')
+    print(f'Потеря сущностей: {ce[1]}')
+    # даём 5 попыток на восстановление
+    for i in range(5):
         regen_prompt = (
         f'При перефразировании текста "{srcText}" из списка элементов "{', '.join(entity for entity in srcEntities)}"'
         f'были утеряны или изменены следующие важные элементы: "{', '.join(e for e in ce[1])}". '
@@ -68,6 +72,10 @@ def validator(srcText, srcEntities, paraEntities):
         newParaphrase = pg.generateByPrompt(regen_prompt)
         paraEntities = ner.extract_entities(newParaphrase)
         ce = compare_entities(srcEntities, paraEntities)
+        print(f'Попытка восстановления: {i+1}')
+        print(f'Исходные сущности: {srcEntities}')
+        print(f'Сгенерируемые сущности: {paraEntities}')
+        print(f'Потеря сущностей: {ce[1]}')
         if (ce[0]): # если сравнение дало True, выходим из цикла
             return newParaphrase
     return None