Update app.py

app.py

@@ -174,6 +174,14 @@ 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
+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 = ""
+
+
 query = st.text_input(
     "Ingrese su búsqueda:",
     placeholder="Ej: Dengue, T630, mordedura de serpiente, intoxicación por mercurio...",
@@ -183,6 +191,9 @@ query = st.text_input(
 if st.button("Buscar", type="primary"):
     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 = ""
     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:
@@ -202,63 +213,63 @@ if st.button("Buscar", type="primary"):
                     protocol_names = []
 
                 if protocol_names:
-                    # Usamos los nombres devueltos por Gemini para hacer una búsqueda directa
                     temp_results = []
                     for name in protocol_names:
                         temp_results.extend(search_direct(name, data_codigos, notificacion_map))
                     
-                    # Eliminar duplicados si Gemini devuelve protocolos que caen en la misma 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)
+            
+            # << CAMBIO 2: GUARDAR RESULTADOS Y CONSULTA EN SESSION STATE >>
+            # En lugar de mostrar los resultados directamente, los guardamos en la "memoria"
+            st.session_state.search_results = results
+            st.session_state.last_query = query
+
+
+# << 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("---")
+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:
+        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}'")
         
-        # 3. Mostrar resultados
-        st.markdown("---")
-        if not results:
-            st.info(f"No se encontraron eventos de notificación obligatoria para su búsqueda: '{query}'.")
-        else:
-            st.success(f"Se encontraron {len(results)} evento(s) relacionado(s) con su búsqueda:")
+        for result in results:
+            info = result["info_notificacion"]
+            codigos = result["detalles_codigos"]
             
-            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):
                 
-                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"
+                # 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', '')}")
-                    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 superinmediata', '')}")
+                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')}")
-                    
-                    # =========================================================================
-                    # ========= INICIO DE LA CORRECCIÓN APLICADA ==============================
-                    # =========================================================================
-                    
-                    # Se saca la operación de reemplazo de la f-string para evitar el SyntaxError
-                    fichas_texto = info.get('Fichas a Utilizar', 'No especificado').replace('+', '\n+ ')
-                    st.markdown(f"**Fichas a utilizar:** {fichas_texto}")
-                    
-                    # =========================================================================
-                    # ========= FIN DE LA CORRECCIÓN APLICADA =================================
-                    # =========================================================================
-                    
-                    # Detalles de códigos
-                    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"**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}")
+                
+                # Detalles de códigos
+                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')}*
+                    """)