Update app.py

app.py

@@ -5,16 +5,18 @@
 
 import streamlit as st
 import google.generativeai as genai
-import google.api_core.exceptions # Import para manejar errores de API
+import google.api_core.exceptions
 import os
 import json
 import logging
 import datetime
+from dotenv import load_dotenv
+load_dotenv()
 
 # --- CONFIGURACIÓN BÁSICA ---
 st.set_page_config(
     page_title="Buscador Inteligente SIVIGILA Colombia",
-    page_icon="buho.png", # Añadido el ícono del búho
+    page_icon="buho.png",
     layout="wide"
 )
 
@@ -24,10 +26,9 @@ 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 en los Secrets de tu Space.")
+        st.error("No se encontró la variable de entorno GEMINI_API_KEY. Por favor, configúrala en tu archivo .env.")
         logger.error("GEMINI_API_KEY no encontrada.")
         st.stop()
     genai.configure(api_key=GEMINI_API_KEY)
@@ -55,21 +56,12 @@ 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.")
         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
@@ -107,54 +99,89 @@ def search_direct(query, data, notif_map):
 
 
 def search_with_gemini(query, event_list):
-    """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 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 EVENTOS OFICIALES:
     {'; '.join(event_list)}
-    
     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 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 API: '{query}'")
         response = model.generate_content(system_prompt)
-        
         if response.parts:
             result_text = response.text.strip()
-            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 API de Gemini: {e}"
 
+# =========================================================================
+# ========= INICIO DE LA FUNCIÓN DE ANÁLISIS ACTUALIZADA ====================
+# =========================================================================
+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.
 
-# --- INTERFAZ DE USUARIO (UI) ---
+    **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.
+
+    **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."
+    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}"
+# =========================================================================
+# ========= FIN DE LA FUNCIÓN DE ANÁLISIS ACTUALIZADA =====================
+# =========================================================================
+
+
+# --- 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_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.")
@@ -177,37 +204,26 @@ 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'])
-                    
+                            if item['Evento'] == name: fichas_encontradas.add(item['FICHA'])
                     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})
-
+                        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)
-
 st.markdown("---")
 
 if st.session_state.search_results is not None:
@@ -220,8 +236,12 @@ 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):
-                tab1, tab2 = st.tabs(["Información General", "📖 Definición de Caso"])
-                with tab1:
+                
+                # 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', '')}")
@@ -263,7 +283,29 @@ 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 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.")
+
+                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'))
+                    else: st.info("No se encontró una definición de caso oficial para este evento.")
+                
+                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..."):
+                            # =========================================================================
+                            # ========= INICIO DE LA LLAMADA A LA FUNCIÓN ACTUALIZADA =================
+                            # =========================================================================
+                            # Extraemos el nombre y la ficha del evento para pasarlos a la función
+                            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)
+                            # =========================================================================
+                            # ========= FIN DE LA LLAMADA A LA FUNCIÓN ACTUALIZADA =====================
+                            # =========================================================================