Update app.py

app.py

@@ -122,10 +122,10 @@ def insert_entity_markers(text: str, entities: List[Dict]) -> str:
     return text
 
 def process_text(text: str) -> Dict[str, List[str]]:
-    """Обрабатывает текст и возвращает сущности и отношения"""
+    """Обрабатывает текст и возвращает сущности и отношения (с повторами, как в pollen_ner.py)"""
     # Получаем сущности
     entities = predict_entities(text)
-    
+
     # Находим отношения
     relations = []
     for i, ent1 in enumerate(entities):
@@ -140,30 +140,24 @@ def process_text(text: str) -> Dict[str, List[str]]:
                             'tail': ent2,
                             'relation': rel
                         })
-    
+
     # Собираем симптомы из отношений has_symptom
-    symptoms = []
+    symptoms_from_rel = []
     for rel in relations:
         if rel['relation'] == 'has_symptom':
             body_part = rel['head']['text'].strip()
             symptom = rel['tail']['text'].strip()
-            # Форматируем симптом как "часть тела + симптом"
-            symptoms.append(f"{body_part} {symptom}")
-    
-    # Группируем сущности по типу (исключая BODY_PART)
+            symptoms_from_rel.append(f"{body_part} {symptom}")
+
+    # Группируем сущности по типу (с повторами)
     ent_by_type = {label: [] for label in ["TOPONYM", "MEDICINE", "SYMPTOM", "ALLERGEN"]}
-    
-    # Добавляем все сущности, кроме SYMPTOM
     for ent in entities:
-        if ent['label'] in ent_by_type and ent['label'] != 'SYMPTOM':
-            # Убираем лишние пробелы и добавляем в список
-            text = ent['text'].strip()
-            if text and text not in ent_by_type[ent['label']]:
-                ent_by_type[ent['label']].append(text)
-    
-    # Добавляем симптомы из отношений
-    ent_by_type['SYMPTOM'] = symptoms
-    
+        if ent['label'] in ent_by_type:
+            ent_by_type[ent['label']].append(ent['text'].strip())
+
+    # Для SYMPTOM: добавляем симптомы из отношений
+    ent_by_type['SYMPTOM'].extend(symptoms_from_rel)
+
     return ent_by_type
 
 def format_output(entities_by_type: Dict[str, List[str]]) -> str:
@@ -242,4 +236,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         debug=True
-    )  
+    )