Update app.py

app.py

@@ -6,7 +6,7 @@
 # \ VVVV/ \VVVV /
 #  \____/ \____/
 # __________________________________________________________________________________________________________
-# Versión Final Estable
+# Versión Final Estable con Búsqueda Específica
 # Adaptado por JAIRO ALEXANDER ERASO MD U Nacional de Colombia, basado en la solicitud del usuario.
 #
 
@@ -69,10 +69,7 @@ def load_data():
         with open(path_notificacion, 'r', encoding='utf-8') as f:
             data_notificacion = json.load(f)
 
-        # Crear un diccionario para búsqueda rápida de datos de notificación por FICHA
         notificacion_map = {item['FICHA']: item for item in data_notificacion}
-        
-        # Generar una lista de nombres de eventos únicos para la búsqueda con IA
         nombres_eventos_unicos = sorted(list(set(item['NOMBRES DE PROTOCOLOS'] for item in data_codigos)))
 
         logger.info(f"✅ Datos cargados: {len(data_codigos)} registros de códigos y {len(data_notificacion)} eventos de notificación.")
@@ -93,34 +90,41 @@ data_codigos, notificacion_map, nombres_eventos_unicos = load_data()
 # --- LÓGICA DE BÚSQUEDA ---
 
 def search_direct(query, data, notif_map):
-    """Realiza una búsqueda directa por código, nombre exacto o ficha."""
+    """
+    Realiza una búsqueda directa y devuelve solo los ítems que coinciden,
+    agrupados por su Ficha/Evento.
+    """
     query_lower = query.lower().strip()
-    results = set() # Usar un set para evitar duplicados de fichas
     
+    # 1. Encontrar todos los ítems individuales que coinciden con la búsqueda
+    matching_items = []
     for item in data:
-        ficha = item.get("FICHA")
-        if not ficha:
-            continue
-
-        # Comprobar coincidencias en varios campos
         if (query_lower == item.get("CIE 10", "").lower().strip() or
             query_lower == item.get("CIE 11", "").lower().strip() or
-            query_lower == ficha.lower().strip() or
+            query_lower == item.get("FICHA", "").lower().strip() or
             query_lower in item.get("NOMBRES DE PROTOCOLOS", "").lower()):
-            results.add(ficha)
+            matching_items.append(item)
             
-    if not results:
+    if not matching_items:
         return []
     
-    # Construir la lista de resultados completos
+    # 2. Agrupar los ítems coincidentes por su número de Ficha
+    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] = []
+            grouped_by_ficha[ficha].append(item)
+            
+    # 3. Construir la lista de resultados final con los datos ya filtrados
     final_results = []
-    for ficha in results:
-        detalles_codigos = [d for d in data if d.get("FICHA") == ficha]
+    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": detalles_codigos
+                "detalles_codigos": matched_details
             })
             
     return final_results
@@ -171,7 +175,6 @@ def search_with_gemini(query, event_list):
 st.title("👨‍⚕️ Asesor Experto 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.")
 
-# Inicialización de session_state para guardar los resultados
 if 'search_results' not in st.session_state:
     st.session_state.search_results = None
 if 'last_query' not in st.session_state:
@@ -182,17 +185,14 @@ def clear_search_state():
     st.session_state.search_results = None
     st.session_state.last_query = ""
 
-# Uso de st.form para agrupar widgets y evitar re-ejecuciones no deseadas
 with st.form(key="search_form"):
     query = st.text_input(
         "Ingrese su búsqueda:",
         placeholder="Ej: Dengue, 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")
 
-# La lógica de búsqueda se ejecuta solo cuando el formulario es enviado
 if submitted:
     if not query:
         st.warning("Por favor, ingrese un término de búsqueda.")
@@ -202,11 +202,8 @@ if submitted:
     else:
         with st.spinner("Buscando..."):
             results = []
-            
-            # 1. Búsqueda directa
             direct_results = search_direct(query, data_codigos, notificacion_map)
             
-            # 2. Si falla, usar IA
             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)
@@ -231,17 +228,14 @@ if submitted:
             else:
                 results = direct_results
 
-            # Guardar resultados en la memoria de la sesión
             st.session_state.search_results = results
             st.session_state.last_query = query
 
-# Botón para limpiar los resultados, visible solo si hay resultados en pantalla
 if st.session_state.search_results is not None:
     st.button("Limpiar Búsqueda", on_click=clear_search_state)
 
 st.markdown("---")
 
-# La lógica de visualización lee siempre de la memoria de la sesión
 if st.session_state.search_results is not None:
     results = st.session_state.search_results
     
@@ -255,7 +249,6 @@ if st.session_state.search_results is not None:
             codigos = result["detalles_codigos"]
             
             with st.expander(f"**Evento: {info.get('Evento', 'Sin nombre')} (Ficha: {info.get('FICHA', 'N/A')})**", expanded=True):
-                
                 st.subheader("Información de Notificación")
                 
                 notif_super = info.get("Notificación superinmedita", "NO") == "SI"