Update app.py

app.py

@@ -167,14 +167,14 @@ SYMPTOM_KEYWORD_MAP = {
 def reinforce_entities_with_keywords(entities, query, food_map, symptom_map):
     """
     Revisa la salida de la IA y añade alimentos Y síntomas clave si fueron omitidos.
-    Esta es una red de seguridad robusta contra las fallas de extracción de la IA.
+    Esta versión es más inteligente para evitar la doble contabilización de síntomas.
     """
     if not entities:
         entities = {"alimentos": [], "sintomas": []}
     
     query_sanitized = sanitize_text(query)
     
-    # --- Reforzar Alimentos (Lógica existente) ---
+    # --- Reforzar Alimentos (Lógica existente, sin cambios) ---
     current_foods = entities.get("alimentos", [])
     current_foods_sanitized = {sanitize_text(f) for f in current_foods}
     for food_keyword in food_map.keys():
@@ -183,15 +183,25 @@ def reinforce_entities_with_keywords(entities, query, food_map, symptom_map):
             current_foods.append(food_keyword)
     entities["alimentos"] = list(set(current_foods))
 
-    # --- NUEVA LÓGICA: Reforzar Síntomas ---
+    # --- LÓGICA MEJORADA: Reforzar Síntomas (Evita redundancia) ---
     current_symptoms = entities.get("sintomas", [])
-    # A veces la IA devuelve None en la lista, lo limpiamos por seguridad
     if current_symptoms is None: current_symptoms = []
-    current_symptoms_sanitized = {sanitize_text(s) for s in current_symptoms}
-    for symptom_keyword in symptom_map.keys():
-        if symptom_keyword in query_sanitized and symptom_keyword not in current_symptoms_sanitized:
+    
+    # Usamos una copia de la consulta para poder modificarla
+    query_to_search_symptoms = query_sanitized
+    
+    # Ordenamos las palabras clave de síntomas por longitud, de la más larga a la más corta
+    # Esto asegura que "dolor de cabeza" se procese antes que "dolor"
+    sorted_symptom_keywords = sorted(symptom_map.keys(), key=len, reverse=True)
+
+    for symptom_keyword in sorted_symptom_keywords:
+        # Si encontramos el síntoma más específico...
+        if symptom_keyword in query_to_search_symptoms:
             logger.info(f"Red de seguridad (Síntoma): Añadiendo '{symptom_keyword}'.")
             current_symptoms.append(symptom_keyword)
+            # ...lo eliminamos de la cadena de búsqueda para que sus sub-partes no se vuelvan a encontrar.
+            query_to_search_symptoms = query_to_search_symptoms.replace(symptom_keyword, "")
+
     entities["sintomas"] = list(set(current_symptoms))
             
     return entities