Update app.py

app.py

@@ -6,6 +6,7 @@
 # \ VVVV/ \VVVV /
 #  \____/ \____/
 # __________________________________________________________________________________________________________
+# Versión Final Estable
 # Adaptado por JAIRO ALEXANDER ERASO MD U Nacional de Colombia, basado en la solicitud del usuario.
 #
 
@@ -114,9 +115,7 @@ def search_direct(query, data, notif_map):
     # Construir la lista de resultados completos
     final_results = []
     for ficha in results:
-        # Encontrar todos los detalles de códigos para esa ficha
         detalles_codigos = [d for d in data if d.get("FICHA") == ficha]
-        # Obtener los datos de notificación
         info_notificacion = notif_map.get(ficha)
         if info_notificacion:
             final_results.append({
@@ -132,7 +131,6 @@ def search_with_gemini(query, event_list):
     if not model:
         return "Error: Modelo de IA no disponible."
 
-    # Prompt cuidadosamente diseñado para obtener una respuesta estructurada
     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.
@@ -158,7 +156,6 @@ def search_with_gemini(query, event_list):
             logger.info(f"✅ Respuesta de Gemini: '{result_text}'")
             if result_text == "NO_ENCONTRADO":
                 return []
-            # Devolver una lista de nombres de protocolos encontrados
             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}")
@@ -174,41 +171,47 @@ 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.")
 
-# << CAMBIO 1: INICIALIZACIÓN DE SESSION STATE >>
-# Nos aseguramos de que las variables existan en la memoria de la sesió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:
     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 = ""
 
-query = st.text_input(
-    "Ingrese su búsqueda:",
-    placeholder="Ej: Dengue, T630, mordedura de serpiente, intoxicación por mercurio...",
-    help="Puede buscar por nombre de la enfermedad, palabras clave, o códigos CIE."
-)
+# 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")
 
-if st.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.")
-        # Limpiamos resultados si la búsqueda está vacía
-        st.session_state.search_results = None
-        st.session_state.last_query = ""
+        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.")
     else:
-        # --- Lógica principal al hacer clic en el botón ---
         with st.spinner("Buscando..."):
+            results = []
             
-            # 1. Intento de búsqueda directa
-            results = search_direct(query, data_codigos, notificacion_map)
+            # 1. Búsqueda directa
+            direct_results = search_direct(query, data_codigos, notificacion_map)
             
-            # 2. Si no hay resultados directos, usar IA
-            if not results:
+            # 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)
                 
-                if isinstance(protocol_names, str): # Manejo de errores de Gemini
+                if isinstance(protocol_names, str):
                     st.error(protocol_names)
                     protocol_names = []
 
@@ -223,18 +226,23 @@ if st.button("Buscar", type="primary"):
                         if ficha not in unique_fichas:
                             results.append(res)
                             unique_fichas.add(ficha)
-            
-            # << CAMBIO 2: GUARDAR RESULTADOS Y CONSULTA EN SESSION STATE >>
-            # En lugar de mostrar los resultados directamente, los guardamos en la "memoria"
+                else:
+                    results = []
+            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)
 
-# << CAMBIO 3: LÓGICA DE VISUALIZACIÓN SEPARADA >>
-# Este bloque se ejecuta en CADA recarga del script. Revisa la "memoria" y muestra lo que hay guardado.
 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:
-    # `is not None` nos permite manejar el caso de una búsqueda que no arrojó resultados (una lista vacía [])
     results = st.session_state.search_results
     
     if not results:
@@ -248,13 +256,11 @@ if st.session_state.search_results is not None:
             
             with st.expander(f"**Evento: {info.get('Evento', 'Sin nombre')} (Ficha: {info.get('FICHA', 'N/A')})**", expanded=True):
                 
-                # Resumen de notificación
                 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 superinmediata', '')}")
+                    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:
@@ -265,7 +271,6 @@ if st.session_state.search_results is not None:
                 fichas_texto = info.get('Fichas a Utilizar', 'No especificado').replace('+', '\n+ ')
                 st.markdown(f"**Fichas a utilizar:** {fichas_texto}")
                 
-                # Detalles de códigos
                 st.subheader("Códigos Relacionados (CIE-10 y CIE-11)")
                 for codigo in codigos:
                     st.markdown(f"""