Update app.py

app.py

@@ -1,11 +1,11 @@
 # ====================    Buscador de Eventos de  Notificación SIVIGILA - Colombia 2025  =====================================
 
-
 # JAIRO ALEXANDER ERASO MD U Nacional de Colombia.
 # DIANA MILENA SOLER MARTINEZ U Juan N. Corpas
 
 import streamlit as st
 import google.generativeai as genai
+import google.api_core.exceptions # Import para manejar errores de API
 import os
 import json
 import logging
@@ -14,7 +14,7 @@ import datetime
 # --- CONFIGURACIÓN BÁSICA ---
 st.set_page_config(
     page_title="Buscador Inteligente SIVIGILA Colombia",
-    page_icon="🇨🇴",
+    page_icon="buho.png", # Añadido el ícono del búho
     layout="wide"
 )
 
@@ -24,9 +24,10 @@ logger = logging.getLogger("sivigila_app")
 
 # --- CONFIGURACIÓN DE GEMINI ---
 try:
+
     GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
     if not GEMINI_API_KEY:
-        st.error("No se encontró la variable de entorno GEMINI_API_KEY. Por favor, configúrala.")
+        st.error("No se encontró la variable de entorno GEMINI_API_KEY. Por favor, configúrala en los Secrets de tu Space.")
         logger.error("GEMINI_API_KEY no encontrada.")
         st.stop()
     genai.configure(api_key=GEMINI_API_KEY)
@@ -38,7 +39,6 @@ except Exception as e:
 
 @st.cache_resource
 def get_gemini_model():
-    """Carga el modelo de Gemini y lo cachea para no recargarlo."""
     logger.info("🔄 Cargando modelo Gemini (gemini-1.5-flash-latest)...")
     try:
         return genai.GenerativeModel("gemini-1.5-flash-latest")
@@ -63,32 +63,27 @@ def load_data():
             data_notificacion = json.load(f)
 
         notificacion_map = {item['FICHA']: item for item in data_notificacion}
-        nombres_eventos_unicos = sorted(list(set(item['NOMBRES DE PROTOCOLOS'] for item in data_codigos)))
+        
+
+        nombres_eventos_limpios = sorted(list(set(item['Evento'] for item in data_notificacion)))
 
         logger.info(f"✅ Datos cargados: {len(data_codigos)} registros de códigos y {len(data_notificacion)} eventos de notificación.")
-        return data_codigos, notificacion_map, nombres_eventos_unicos
+        return data_codigos, data_notificacion, notificacion_map, nombres_eventos_limpios
 
     except FileNotFoundError as e:
         st.error(f"❌ Error: No se encontró el archivo {e.filename}. Asegúrate de que los archivos JSON estén en la carpeta 'PLANTILLAS'.")
-        logger.error(f"Archivo no encontrado: {e.filename}")
-        return None, None, None
+        return None, None, None, None
     except json.JSONDecodeError as e:
         st.error(f"❌ Error: El archivo JSON tiene un formato incorrecto: {e}")
-        logger.error(f"Error de formato JSON: {e}")
-        return None, None, None
+        return None, None, None, None
 
-data_codigos, notificacion_map, nombres_eventos_unicos = load_data()
+data_codigos, data_notificacion, notificacion_map, nombres_eventos_limpios = load_data()
 
 
 # --- LÓGICA DE BÚSQUEDA ---
 
 def search_direct(query, data, notif_map):
-    """
-    Realiza una búsqueda directa y devuelve solo los ítems que coinciden,
-    agrupados por su Ficha/Evento.
-    """
     query_lower = query.lower().strip()
-    
     matching_items = []
     for item in data:
         if (query_lower == item.get("CIE 10", "").lower().strip() or
@@ -96,249 +91,179 @@ def search_direct(query, data, notif_map):
             query_lower == item.get("FICHA", "").lower().strip() or
             query_lower in item.get("NOMBRES DE PROTOCOLOS", "").lower()):
             matching_items.append(item)
-            
-    if not matching_items:
-        return []
-    
+    if not matching_items: return []
     grouped_by_ficha = {}
     for item in matching_items:
         ficha = item.get("FICHA")
         if ficha:
-            if ficha not in grouped_by_ficha:
-                grouped_by_ficha[ficha] = []
+            if ficha not in grouped_by_ficha: grouped_by_ficha[ficha] = []
             grouped_by_ficha[ficha].append(item)
-            
     final_results = []
     for ficha, matched_details in grouped_by_ficha.items():
         info_notificacion = notif_map.get(ficha)
         if info_notificacion:
-            final_results.append({
-                "info_notificacion": info_notificacion,
-                "detalles_codigos": matched_details
-            })
-            
+            final_results.append({"info_notificacion": info_notificacion, "detalles_codigos": matched_details})
     return final_results
 
 
 def search_with_gemini(query, event_list):
-    """Usa Gemini para encontrar el evento más relevante a partir de una consulta en lenguaje natural."""
-    if not model:
-        return "Error: Modelo de IA no disponible."
+    """Usa la API de Gemini para encontrar el evento más relevante."""
+    if not model: return "Error: Modelo Gemini no disponible."
+
 
     system_prompt = f"""
     Eres un asesor experto en SIVIGILA, el sistema de vigilancia en salud pública de Colombia.
-    Tu tarea es identificar el evento de notificación más relevante basado en la consulta de un profesional de la salud.
-    A continuación, te proporciono una lista de los nombres oficiales de los protocolos de notificación.
+    Tu tarea es interpretar la consulta de un profesional de la salud, que puede ser un término coloquial, un sinónimo o una descripción, y asociarla al nombre del evento oficial más relevante de una lista.
     
-    LISTA DE PROTOCOLOS OFICIALES:
+    LISTA DE EVENTOS OFICIALES:
     {'; '.join(event_list)}
     
-    Analiza la siguiente consulta del usuario y devuelve ÚNICAMENTE el nombre exacto del protocolo o protocolos de la lista que mejor correspondan.
+    Analiza la siguiente consulta y devuelve ÚNICAMENTE el nombre exacto del evento oficial de la lista que mejor corresponda.
+    Por ejemplo, si la consulta es 'mordedura de culebra', la respuesta debe ser 'Accidente Ofídico'.
     Si encuentras varios, sepáralos con un punto y coma (;).
-    Si la consulta no corresponde a ningún protocolo de la lista, responde con "NO_ENCONTRADO".
-    No añadas explicaciones, saludos ni texto adicional. Solo el nombre del protocolo.
+    Si la consulta no corresponde a ningún evento de la lista, responde con "NO_ENCONTRADO".
+    No añadas explicaciones ni texto adicional.
     
     Consulta del usuario: "{query}"
     """
+
     
     try:
-        logger.info(f"🔄 Enviando consulta a Gemini: '{query}'")
+        logger.info(f"🔄 Enviando consulta a Gemini API: '{query}'")
         response = model.generate_content(system_prompt)
         
         if response.parts:
             result_text = response.text.strip()
-            logger.info(f"✅ Respuesta de Gemini: '{result_text}'")
-            if result_text == "NO_ENCONTRADO":
-                return []
+            logger.info(f"✅ Respuesta de Gemini API: '{result_text}'")
+            if result_text == "NO_ENCONTRADO": return []
             return [name.strip() for name in result_text.split(';') if name.strip()]
         else:
             logger.warning(f"⚠️ Respuesta de Gemini vacía o bloqueada. Razón: {response.prompt_feedback}")
             return "Respuesta bloqueada por filtros de seguridad."
             
+    except google.api_core.exceptions.ResourceExhausted as e:
+        logger.warning(f"Límite de frecuencia de la API alcanzado: {e}")
+        return "Se ha alcanzado el límite de consultas. Por favor, espere un minuto y vuelva a intentarlo."
     except Exception as e:
         logger.error(f"❌ Error en la llamada a Gemini: {e}")
-        return f"Error en la comunicación con la IA: {e}"
+        return f"Error en la comunicación con la API de Gemini: {e}"
 
 
 # --- INTERFAZ DE USUARIO (UI) ---
 
-st.title("👨‍⚕️ Buscador inteligente en Notificación SIVIGILA - Colombia")
-st.markdown("Busque una enfermedad por nombre, sinónimo, código CIE-10 o CIE-11 para obtener información de notificación.")
 
-if 'search_results' not in st.session_state:
-    st.session_state.search_results = None
-if 'last_query' not in st.session_state:
-    st.session_state.last_query = ""
+col_img, col_text = st.columns([1, 5], gap="medium")
+with col_img:
+    st.image("buho.png", width=100)
+with col_text:
+    st.title("Buscador Inteligente SIVIGILA")
+    st.markdown("Herramienta de apoyo para la notificación de eventos de salud pública en Colombia.")
+
+if 'search_results' not in st.session_state: st.session_state.search_results = None
+if 'last_query' not in st.session_state: st.session_state.last_query = ""
 
 def clear_search_state():
-    """Función para reiniciar el estado de la búsqueda."""
     st.session_state.search_results = None
     st.session_state.last_query = ""
 
 with st.form(key="search_form"):
-    query = st.text_input(
-        "Ingrese su búsqueda:",
-        placeholder="Ej: Lepra, T630, mordedura de serpiente, 3MC...",
-        help="Puede buscar por nombre de la enfermedad, palabras clave, o códigos CIE."
-    )
+    query = st.text_input("Ingrese su búsqueda:", placeholder="Ej: Lepra, T630, mordedura de serpiente, 3MC...", help="Puede buscar por nombre de la enfermedad, palabras clave, o códigos CIE.")
     submitted = st.form_submit_button("Buscar", type="primary")
 
 if submitted:
-    if not query:
-        st.warning("Por favor, ingrese un término de búsqueda.")
-        clear_search_state()
-    elif data_codigos is None or notificacion_map is None:
-        st.error("Los datos no se pudieron cargar. La aplicación no puede funcionar.")
+    if not query: st.warning("Por favor, ingrese un término de búsqueda.")
+    elif data_codigos is None: st.error("Los datos no se pudieron cargar.")
     else:
-        with st.spinner("Buscando..."):
+        with st.spinner("Buscando con IA (Gemini)..."):
             results = []
             direct_results = search_direct(query, data_codigos, notificacion_map)
             
             if not direct_results:
-                logger.info(f"Búsqueda directa para '{query}' no arrojó resultados. Usando Gemini...")
-                protocol_names = search_with_gemini(query, nombres_eventos_unicos)
+                logger.info(f"Búsqueda directa para '{query}' no arrojó resultados. Usando IA (Gemini)...")
+                event_names = search_with_gemini(query, nombres_eventos_limpios)
                 
-                if isinstance(protocol_names, str):
-                    st.error(protocol_names)
-                    protocol_names = []
-
-                if protocol_names:
-                    temp_results = []
-                    for name in protocol_names:
-                        temp_results.extend(search_direct(name, data_codigos, notificacion_map))
+                if isinstance(event_names, str):
+                    st.error(event_names)
+                    event_names = []
+
+
+                if event_names:
+                    fichas_encontradas = set()
+                    for name in event_names:
+                        for item in data_notificacion:
+                            if item['Evento'] == name:
+                                fichas_encontradas.add(item['FICHA'])
                     
-                    unique_fichas = set()
-                    for res in temp_results:
-                        ficha = res["info_notificacion"]["FICHA"]
-                        if ficha not in unique_fichas:
-                            results.append(res)
-                            unique_fichas.add(ficha)
-                else:
-                    results = []
+                    for ficha in fichas_encontradas:
+                        info_notif = notificacion_map.get(ficha)
+                        detalles_cods = [d for d in data_codigos if d.get("FICHA") == ficha]
+                        if info_notif:
+                            results.append({"info_notificacion": info_notif, "detalles_codigos": detalles_cods})
+
             else:
                 results = direct_results
 
             st.session_state.search_results = results
             st.session_state.last_query = query
 
-if st.session_state.search_results is not None:
-    st.button("Limpiar Búsqueda", on_click=clear_search_state)
+if st.session_state.search_results is not None: st.button("Limpiar Búsqueda", on_click=clear_search_state)
 
 st.markdown("---")
 
 if st.session_state.search_results is not None:
     results = st.session_state.search_results
-    
     if not results:
         st.info(f"No se encontraron eventos de notificación obligatoria para su búsqueda: '{st.session_state.last_query}'.")
     else:
         st.success(f"Se encontraron {len(results)} evento(s) relacionado(s) con su búsqueda: '{st.session_state.last_query}'")
-        
         for result in results:
             info = result["info_notificacion"]
             codigos = result["detalles_codigos"]
-            
             with st.expander(f"**Evento: {info.get('Evento', 'Sin nombre')} (Ficha: {info.get('FICHA', 'N/A')})**", expanded=True):
-                
-                
                 tab1, tab2 = st.tabs(["Información General", "📖 Definición de Caso"])
-
                 with tab1:
                     st.subheader("Información de Notificación")
-                    
                     notif_super = info.get("Notificación superinmedita", "NO") == "SI"
-                    if notif_super:
-                        st.error(f"**¡ATENCIÓN! Requiere Notificación SUPERINMEDIATA:** {info.get('Descripción superinmedita', '')}")
-                    elif info.get("Notificación Inmediata", "NO") == "SI":
-                        st.warning("**Requiere Notificación Inmediata (dentro de las 24 horas).**")
-                    else:
-                        st.info("**Notificación Semanal Individual.**")
-
+                    if notif_super: st.error(f"**¡ATENCIÓN! Requiere Notificación SUPERINMEDIATA:** {info.get('Descripción superinmedita', '')}")
+                    elif info.get("Notificación Inmediata", "NO") == "SI": st.warning("**Requiere Notificación Inmediata (dentro de las 24 horas).**")
+                    else: st.info("**Notificación Semanal Individual.**")
                     st.markdown(f"**Requisito:** {info.get('REQUISITO', 'No especificado')}")
-                    
                     fichas_texto = info.get('Fichas a Utilizar', 'No especificado').replace('+', '\n+ ')
                     st.markdown(f"**Fichas a utilizar:** {fichas_texto}")
-
-                    st.markdown("---") 
+                    st.markdown("---")
                     st.subheader("Clasificación de Caso Permitida")
-                    
-                    classification_fields = [
-                        ("Sospechoso", "Clasificación Permitida Sospechoso"),
-                        ("Probable", "Clasificación Permitida Probable"),
-                        ("Conf. Clínica", "Clasificación Permitida Conf. Clínica"),
-                        ("Conf. Laboratorio", "Clasificación Permitida Conf. Laboratorio"),
-                        ("Conf. Nexo Epi.", "Clasificación Permitida Conf. Nexo Epi."),
-                        ("Descartado", "Clasificación Permitida Descartado")
-                    ]
-                    
-                    col1, col2 = st.columns(2)
-                    
-                    for i, (display_name, field_key) in enumerate(classification_fields):
-                        value = info.get(field_key, "NO")
-                        icon = "✅" if value == "SI" else "❌"
-                        md_string = f"{icon} **{display_name}**"
-                        
-                        if i < len(classification_fields) / 2:
-                            col1.markdown(md_string)
-                        else:
-                            col2.markdown(md_string)
-
+                    classification_fields=[("Sospechoso","Clasificación Permitida Sospechoso"),("Probable","Clasificación Permitida Probable"),("Conf. Clínica","Clasificación Permitida Conf. Clínica"),("Conf. Laboratorio","Clasificación Permitida Conf. Laboratorio"),("Conf. Nexo Epi.","Clasificación Permitida Conf. Nexo Epi."),("Descartado","Clasificación Permitida Descartado")]
+                    col1,col2=st.columns(2)
+                    for i,(display_name,field_key) in enumerate(classification_fields):
+                        value=info.get(field_key,"NO")
+                        icon="✅" if value=="SI" else "❌"
+                        md_string=f"{icon} **{display_name}**"
+                        if i<len(classification_fields)/2: col1.markdown(md_string)
+                        else: col2.markdown(md_string)
                     st.markdown("---")
                     st.subheader("Rango de Edad Permitido para Notificación")
-
-                    age_fields = [
-                        ("Menor de 1 año", "Menor de 1 año"),
-                        ("De 1 a 4 años", "De 1 a 4 años"),
-                        ("De 5 a 14 años", "De 5 a 14 años"),
-                        ("De 15 a 44 años", "De 15 a 44 años"),
-                        ("De 45 a 59 años", "De 45 a 59 años"),
-                        ("De 60 y más", "De 60 y más")
-                    ]
-                    
-                    age_col1, age_col2 = st.columns(2)
-
-                    for i, (display_name, field_key) in enumerate(age_fields):
-                        value = info.get(field_key, "NO")
-                        icon = "✅" if value == "SI" else "❌"
-                        md_string = f"{icon} **{display_name}**"
-                        
-                        if i < len(age_fields) / 2:
-                            age_col1.markdown(md_string)
-                        else:
-                            age_col2.markdown(md_string)
-
+                    age_fields=[("Menor de 1 año","Menor de 1 año"),("De 1 a 4 años","De 1 a 4 años"),("De 5 a 14 años","De 5 a 14 años"),("De 15 a 44 años","De 15 a 44 años"),("De 45 a 59 años","De 45 a 59 años"),("De 60 y más","De 60 y más")]
+                    age_col1,age_col2=st.columns(2)
+                    for i,(display_name,field_key) in enumerate(age_fields):
+                        value=info.get(field_key,"NO")
+                        icon="✅" if value=="SI" else "❌"
+                        md_string=f"{icon} **{display_name}**"
+                        if i<len(age_fields)/2: age_col1.markdown(md_string)
+                        else: age_col2.markdown(md_string)
                     st.markdown("---")
                     st.subheader("Condición Final Permitida")
-                    
-                    final_condition_fields = [
-                        ("Vivo", "Condición final Vivo"),
-                        ("Muerto", "Condición final Muerto")
-                    ]
-
-                    cond_col1, cond_col2 = st.columns(2)
-
-                    for i, (display_name, field_key) in enumerate(final_condition_fields):
-                        value = info.get(field_key, "NO")
-                        icon = "✅" if value == "SI" else "❌"
-                        md_string = f"{icon} **{display_name}**"
-                        
-                        if i == 0:
-                            cond_col1.markdown(md_string)
-                        else:
-                            cond_col2.markdown(md_string)
-
+                    final_condition_fields=[("Vivo","Condición final Vivo"),("Muerto","Condición final Muerto")]
+                    cond_col1,cond_col2=st.columns(2)
+                    for i,(display_name,field_key) in enumerate(final_condition_fields):
+                        value=info.get(field_key,"NO")
+                        icon="✅" if value=="SI" else "❌"
+                        md_string=f"{icon} **{display_name}**"
+                        if i==0: cond_col1.markdown(md_string)
+                        else: cond_col2.markdown(md_string)
                     st.subheader("Códigos Relacionados (CIE-10 y CIE-11)")
                     for codigo in codigos:
-                        st.markdown(f"""
-                        - **Nombre Protocolo:** `{codigo.get('NOMBRES DE PROTOCOLOS', 'N/A')}`
-                          - **CIE-10:** `{codigo.get('CIE 10', 'N/A')}`
-                          - **CIE-11:** `{codigo.get('CIE 11', 'N/A')}` | **Nombre CIE-11:** *{codigo.get('NOMBRES DE CIE 11', 'N/A')}*
-                        """)
-                
+                        st.markdown(f"""- **Nombre Protocolo:** `{codigo.get('NOMBRES DE PROTOCOLOS', 'N/A')}`\n  - **CIE-10:** `{codigo.get('CIE 10', 'N/A')}`\n  - **CIE-11:** `{codigo.get('CIE 11', 'N/A')}` | **Nombre CIE-11:** *{codigo.get('NOMBRES DE CIE 11', 'N/A')}*""")
                 with tab2:
-                    definicion_de_caso = info.get("Definición de caso", "").strip()
-                    if definicion_de_caso:
-                        st.markdown(definicion_de_caso.replace('\n', '  \n'))
-                    else:
-                        st.info("No se encontró una definición de caso oficial para este evento.")
-                
+                    definicion_de_caso=info.get("Definición de caso","").strip()
+                    if definicion_de_caso: st.markdown(definicion_de_caso.replace('\n','  \n'))
+                    else: st.info("No se encontró una definición de caso oficial para este evento.")