Update app.py

app.py

@@ -261,7 +261,36 @@ FOOD_NAME_TO_FOODB_KEY = {
     "vino": ["wine", "red wine", "white wine"]
 }
 
+# --- AÑADE ESTA FUNCIÓN NUEVA EN TU CÓDIGO ---
 
+def reinforce_entities_with_keywords(entities, query, food_map):
+    """
+    Revisa la salida de la IA y añade alimentos clave si fueron omitidos.
+    Esta es una red de seguridad contra las fallas de extracción de la IA.
+    """
+    if not entities:
+        entities = {"alimentos": [], "sintomas": []}
+    
+    query_sanitized = sanitize_text(query)
+    current_foods = entities.get("alimentos", [])
+    
+    # Si la IA ya encontró alimentos, los sanitizamos para la comparación
+    current_foods_sanitized = {sanitize_text(f) for f in current_foods}
+    
+    # Iteramos por nuestra lista de alimentos conocidos
+    for food_keyword in food_map.keys():
+        # Si una palabra clave de alimento está en la consulta...
+        if food_keyword in query_sanitized:
+            # ...y la IA NO la incluyó en su lista...
+            if food_keyword not in current_foods_sanitized:
+                # ...la añadimos nosotros.
+                logger.info(f"Red de seguridad: La IA omitió '{food_keyword}', añadiéndolo manualmente.")
+                current_foods.append(food_keyword)
+                
+    entities["alimentos"] = list(set(current_foods)) # Eliminar duplicados
+    return entities
+
+    
 # --- LÓGICA DE BÚSQUEDA Y ANÁLISIS ---
 # (Todas las funciones auxiliares se mantienen igual que en la versión anterior)
 def sanitize_text(text):
@@ -501,6 +530,8 @@ if submitted:
     else:
         with st.spinner("🧠 Interpretando tu caso y buscando pistas con IA..."):
             entities = extract_and_infer_with_gemini(query, lista_condiciones)
+            # --- AÑADE ESTA LÍNEA PARA ACTIVAR LA RED DE SEGURIDAD ---
+            entities = reinforce_entities_with_keywords(entities, query, FOOD_TO_COMPOUND_MAP)
             st.session_state.entities = entities
         if entities and (entities.get("alimentos") or entities.get("sintomas")):
             info_str = f"IA identificó - Alimentos: {', '.join(entities.get('alimentos',[])) or 'Ninguno'}, Síntomas: {', '.join(entities.get('sintomas',[])) or 'Ninguno'}"