Update app.py

app.py

@@ -1,4 +1,5 @@
-# ====================    Buscador de Eventos para  Notificación SIVIGILA - Colombia 2025  =====================================
+# ====================    Buscador de Eventos de  Notificación SIVIGILA - Colombia 2025  =====================================
+# Versión API Web Final con Lógica de IA Mejorada y Segundo Filtro
 
 # JAIRO ALEXANDER ERASO MD U Nacional de Colombia.
 # DIANA MILENA SOLER MARTINEZ U Juan N. Corpas
@@ -12,8 +13,9 @@ import logging
 import datetime
 
 
+# --- CONFIGURACIÓN BÁSICA ---
 st.set_page_config(
-    page_title="Buscador Inteligente de eventos para notificación SIVIGILA Colombia",
+    page_title="Buscador Inteligente SIVIGILA Colombia",
     page_icon="buho.png",
     layout="wide"
 )
@@ -26,7 +28,7 @@ logger = logging.getLogger("sivigila_app")
 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 en tu archivo .env.")
+        st.error("No se encontró la variable de entorno GEMINI_API_KEY. Por favor, configúrala en tu archivo .env o en los Secrets.")
         logger.error("GEMINI_API_KEY no encontrada.")
         st.stop()
     genai.configure(api_key=GEMINI_API_KEY)
@@ -54,12 +56,19 @@ def load_data():
     try:
         path_codigos = os.path.join('PLANTILLAS', 'DIC_NOMBRES_CODIGOS_FICHAS.json')
         path_notificacion = os.path.join('PLANTILLAS', 'DATOS_NOTIFICACION.json')
-        with open(path_codigos, 'r', encoding='utf-8') as f: data_codigos = json.load(f)
-        with open(path_notificacion, 'r', encoding='utf-8') as f: data_notificacion = json.load(f)
+
+        with open(path_codigos, 'r', encoding='utf-8') as f:
+            data_codigos = json.load(f)
+        
+        with open(path_notificacion, 'r', encoding='utf-8') as f:
+            data_notificacion = json.load(f)
+
         notificacion_map = {item['FICHA']: item for item in data_notificacion}
         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.")
+
+        logger.info(f"✅ Datos cargados.")
         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'.")
         return None, None, None, None
@@ -125,57 +134,43 @@ def search_with_gemini(query, event_list):
 
 
 def analyze_query_with_gemini(query, definition, evento_name, ficha_number):
-    """
-    Usa la API de Gemini para analizar la consulta y emitir una recomendación
-    prescriptiva sobre la notificación.
-    """
     if not model: return "Error: Modelo Gemini no disponible."
-
     analysis_prompt = f"""
     Eres un auditor experto del sistema SIVIGILA de Colombia. Tu única misión es determinar si un caso descrito por un profesional de la salud cumple con los criterios de la definición de caso oficial para ser de notificación obligatoria.
-
     **Reglas Estrictas:**
-    1.  Tu decisión final solo puede ser "NOTIFICAR" o "NO CUMPLE CRITERIOS".
-    2.  Debes basar tu justificación únicamente en la comparación entre la consulta y la definición de caso proporcionada.
-    3.  No debes inventar información ni hacer suposiciones clínicas.
-
+    1. Tu decisión final solo puede ser "NOTIFICAR" o "NO CUMPLE CRITERIOS".
+    2. Debes basar tu justificación únicamente en la comparación entre la consulta y la definición de caso proporcionada.
+    3. No debes inventar información ni hacer suposiciones clínicas.
     **CONTEXTO DEL EVENTO YA IDENTIFICADO:**
     - Nombre del Evento: "{evento_name}"
     - Ficha de Notificación: "{ficha_number}"
-
     **CONSULTA DEL USUARIO:**
     "{query}"
-
     **DEFINICIÓN DE CASO OFICIAL:**
     "{definition}"
-
     **TU ANÁLISIS ESTRUCTURADO:**
     Basado en la comparación, responde OBLIGATORIAMENTE en el siguiente formato, sin añadir texto adicional:
-
     **Decisión:** [Escribe aquí "NOTIFICAR" o "NO CUMPLE CRITERIOS"]
-
     **Justificación:**
     [Si la decisión es NOTIFICAR, explica brevemente qué criterios de la definición de caso se cumplen en la consulta del usuario. Si la decisión es NO CUMPLE CRITERIOS, explica qué criterios clave faltan o se contradicen explícitamente.]
-
     **Recomendación:**
     [Si la decisión es NOTIFICAR, redacta una recomendación para proceder con la notificación, mencionando explícitamente el número de Ficha y el nombre del Evento proporcionados en el contexto. Si la decisión es NO CUMPLE CRITERIOS, escribe: "Se recomienda revisar las guías y protocolos del Instituto Nacional de Salud de Colombia para determinar el manejo adecuado del caso."]
     """
     try:
         response = model.generate_content(analysis_prompt)
-        if response.parts:
-            return response.text
-        else:
-            return "El análisis no pudo ser completado. La respuesta de la IA fue bloqueada por filtros de seguridad."
+        if response.parts: return response.text
+        else: return "El análisis no pudo ser completado. La respuesta de la IA fue bloqueada por filtros de seguridad."
     except google.api_core.exceptions.ResourceExhausted:
         return "Se ha alcanzado el límite de consultas para el análisis. Por favor, espere un minuto y vuelva a intentarlo."
     except Exception as e:
         return f"Ocurrió un error durante el análisis de la IA: {e}"
 
+
 # --- INTERFAZ DE USUARIO (UI) ---
 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 para notificar SIVIGILA CIE 10-11")
+    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
@@ -186,7 +181,7 @@ def clear_search_state():
     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, Niño 3 años con 9 kilos, 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:
@@ -196,22 +191,44 @@ if submitted:
         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 IA (Gemini)...")
                 event_names = search_with_gemini(query, nombres_eventos_limpios)
+                
                 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'])
+                    
+                    query_keywords = [kw for kw in query.lower().split() if len(kw) > 2] # Keywords de la búsqueda original
+
                     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})
+                        all_details_for_ficha = [d for d in data_codigos if d.get("FICHA") == ficha]
+                        
+                        filtered_details = []
+                        if query_keywords:
+                            for detail in all_details_for_ficha:
+                                protocol_name_lower = detail.get("NOMBRES DE PROTOCOLOS", "").lower()
+                                if any(keyword in protocol_name_lower for keyword in query_keywords):
+                                    filtered_details.append(detail)
+                        
+                        final_details = filtered_details if filtered_details else all_details_for_ficha
+                        
+                        if info_notif:
+                            results.append({
+                                "info_notificacion": info_notif,
+                                "detalles_codigos": final_details
+                            })
             else:
                 results = direct_results
+
             st.session_state.search_results = results
             st.session_state.last_query = query
 
@@ -228,12 +245,8 @@ if st.session_state.search_results is not None:
             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):
-                
-                # Se mantienen las 3 pestañas
                 tab_info, tab_def, tab_analysis = st.tabs(["Información General", "📖 Definición de Caso", "🤖 Análisis por IA"])
-
                 with tab_info:
-                    # El contenido de esta pestaña no cambia
                     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', '')}")
@@ -275,7 +288,7 @@ if st.session_state.search_results is not None:
                     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')}`\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 tab_def:
                     definicion_de_caso = info.get("Definición de caso", "").strip()
                     if definicion_de_caso: st.markdown(definicion_de_caso.replace('\n', '  \n'))
@@ -284,15 +297,12 @@ if st.session_state.search_results is not None:
                 with tab_analysis:
                     st.info("Este análisis compara su búsqueda original con la definición oficial del evento encontrado.")
                     definicion_de_caso_para_analisis = info.get("Definición de caso", "").strip()
-                    
                     if not definicion_de_caso_para_analisis:
                         st.warning("No se puede realizar el análisis porque no hay una definición de caso disponible para este evento.")
                     else:
-                        with st.spinner("Realizando análisis con IA..."):
 
-                            # Extraemos el nombre y la ficha del evento para pasarlos a la función
+                        with st.spinner("Realizando análisis con IA..."):
                             evento_name = info.get("Evento", "N/A")
                             ficha_number = info.get("FICHA", "N/A")
-                            
                             analysis_result = analyze_query_with_gemini(st.session_state.last_query, definicion_de_caso_para_analisis, evento_name, ficha_number)
                             st.markdown(analysis_result)