Update app.py

app.py

@@ -151,18 +151,49 @@ FOOD_NAME_TO_FOODB_KEY = {
     "aceituna": ["olive"], "café": ["coffee"], "caldo": ["broth", "stock"], "cerveza": ["beer"], "chocolate": ["chocolate"], "jengibre": ["ginger"], "cúrcuma": ["turmeric"], "miel": ["honey"], "mostaza": ["mustard"], "té": ["tea"], "vinagre": ["vinegar"], "vino": ["wine", "red wine", "white wine"]
 }
 
-def reinforce_entities_with_keywords(entities, query, food_map):
+SYMPTOM_KEYWORD_MAP = {
+    "dolor de cabeza": ["dolor de cabeza", "cefalea"],
+    "hinchazón": ["hinchazón", "distensión", "inflamado"],
+    "gases": ["gases", "flatulencia"],
+    "diarrea": ["diarrea"],
+    "dolor": ["dolor"],
+    "fatiga": ["fatiga", "cansancio"],
+    "náuseas": ["náuseas", "mareo"],
+    "vómitos": ["vómitos"],
+    "erupción": ["erupción", "ronchas", "urticaria"],
+    "acidez": ["acidez", "reflujo", "ardor de estómago"]
+}
+
+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.
+    """
     if not entities:
         entities = {"alimentos": [], "sintomas": []}
+    
     query_sanitized = sanitize_text(query)
+    
+    # --- Reforzar Alimentos (Lógica existente) ---
     current_foods = entities.get("alimentos", [])
     current_foods_sanitized = {sanitize_text(f) for f in current_foods}
     for food_keyword in food_map.keys():
-        if food_keyword in query_sanitized:
-            if food_keyword not in current_foods_sanitized:
-                logger.info(f"Red de seguridad: La IA omitió '{food_keyword}', añadiéndolo manualmente.")
-                current_foods.append(food_keyword)
+        if food_keyword in query_sanitized and food_keyword not in current_foods_sanitized:
+            logger.info(f"Red de seguridad (Alimento): Añadiendo '{food_keyword}'.")
+            current_foods.append(food_keyword)
     entities["alimentos"] = list(set(current_foods))
+
+    # --- NUEVA LÓGICA: Reforzar Síntomas ---
+    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:
+            logger.info(f"Red de seguridad (Síntoma): Añadiendo '{symptom_keyword}'.")
+            current_symptoms.append(symptom_keyword)
+    entities["sintomas"] = list(set(current_symptoms))
+            
     return entities
 
 def sanitize_text(text):
@@ -405,7 +436,7 @@ if st.session_state.start_analysis:
     else:
         with st.spinner("🧠 Interpretando tu caso y buscando pistas con IA..."):
             entities = extract_and_infer_with_gemini(query_to_analyze, lista_condiciones)
-            entities = reinforce_entities_with_keywords(entities, query_to_analyze, FOOD_TO_COMPOUND_MAP)
+            entities = reinforce_entities_with_keywords(entities, query_to_analyze, FOOD_TO_COMPOUND_MAP, SYMPTOM_KEYWORD_MAP)
             st.session_state.entities = entities
         
         if entities and (entities.get("alimentos") or entities.get("sintomas")):