Update app.py

app.py

@@ -300,7 +300,7 @@ def buscar_datos_guardados():
 
                 # Link to download the file
                 try:
-                    with open(os.path.join(carpeta, archivo_seleccionado), 'rb') as file):
+                    with open(os.path.join(carpeta, archivo_seleccionado), 'rb') as file:
                         st.download_button(
                             label="📥 Descargar Archivo TXT",
                             data=file,
@@ -457,6 +457,129 @@ def mostrar_recomendaciones():
     st.title("⭐ Recomendaciones")
     st.write("Aquí puedes encontrar recomendaciones y consejos útiles.")
 
+def main():
+    st.set_page_config(page_title="Galatea OMARDENT", layout="wide")
+    
+    # Inicializar el estado de la sesión
+    if 'modelo' not in st.session_state:
+        st.session_state['modelo'] = "gpt-3.5-turbo"
+    if 'temperatura' not in st.session_state:
+        st.session_state['temperatura'] = 0.5
+    if 'mensajes_chat' not in st.session_state:
+        st.session_state['mensajes_chat'] = []
+    if 'transcripcion_voz' not in st.session_state:
+        st.session_state['transcripcion_voz'] = ""
+    if 'imagen_asistente' not in st.session_state:
+        st.session_state['imagen_asistente'] = None
+    if 'video_estado' not in st.session_state:
+        st.session_state['video_estado'] = 'paused'
+    if 'assistant_id' not in st.session_state:
+        st.session_state['assistant_id'] = 'asst_4ZYvBvf4IUVQPjnugSZGLdV2'
+    if 'presupuesto_texto' not in st.session_state:
+        st.session_state['presupuesto_texto'] = ''
+    if 'mostrar_chat' not in st.session_state:
+        st.session_state['mostrar_chat'] = False
+    if 'texto_pdf' not in st.session_state:
+        with open("assets/instrucciones.pdf", "rb") as file:
+            st.session_state['texto_pdf'] = extraer_texto_pdf(file)
+
+    # Barra lateral
+    ruta_logo = os.path.join("assets", "Logo Omardent.png")
+    if os.path.exists(ruta_logo):
+        st.sidebar.image(ruta_logo, use_column_width=True)
+    else:
+        st.sidebar.warning(f"Error: No se pudo encontrar la imagen en la ruta: {ruta_logo}")
+
+    st.sidebar.title("🤖 Galatea OMARDENT")
+    st.sidebar.markdown("---")
+    st.sidebar.subheader("🧠 Configuración del Modelo")
+    st.session_state['modelo'] = st.sidebar.selectbox(
+        "Selecciona el modelo:",
+        ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k", "gpt-4o"],
+        index=0,
+        key='modelo_selectbox',  # Clave única
+        help="Elige el modelo de lenguaje de OpenAI que prefieras."
+    )
+    st.sidebar.markdown("---")
+    st.session_state['temperatura'] = st.sidebar.slider(
+        "🌡️ Temperatura",
+        min_value=0.0, max_value=1.0,
+        value=st.session_state['temperatura'],
+        step=0.1,
+        key='temperatura_slider'  # Clave única
+    )
+    assistant_id = st.sidebar.text_input("Assistant ID", key="assistant_id", help="Introduce el Assistant ID del playground de OpenAI")
+    
+    st.sidebar.markdown("---")
+    st.sidebar.subheader("🌟 Navegación")
+    lateral_page = st.sidebar.radio("Ir a", ["Página Principal", "Gestión de Trabajos", "Gestión de Insumos", "Registro de Radiografías", "Buscar Datos", "Notificaciones", "Recomendaciones", "Asistente de Presupuestos", "Comunicación", "Asistente de Agendamiento"])
+
+    top_page = st.selectbox("Navegación Superior", ["Página Principal", "Galatea-Asistente"])
+
+    if top_page == "Galatea-Asistente":
+        mostrar_galatea_asistente()
+    else:
+        if lateral_page == "Página Principal":
+            mostrar_pagina_principal()
+        elif lateral_page == "Gestión de Trabajos":
+            flujo_laboratorio()
+        elif lateral_page == "Gestión de Insumos":
+            flujo_insumos()
+        elif lateral_page == "Registro de Radiografías":
+            flujo_radiografias()
+        elif lateral_page == "Buscar Datos":
+            buscar_datos_guardados()
+        elif lateral_page == "Notificaciones":
+            generar_notificaciones_pendientes()
+        elif lateral_page == "Recomendaciones":
+            mostrar_recomendaciones()
+        elif lateral_page == "Asistente de Presupuestos":
+            flujo_presupuestos()
+        elif lateral_page == "Comunicación":
+            st.write("Página de Comunicación")  # Implementar según sea necesario
+        elif lateral_page == "Asistente de Agendamiento":
+            st.write("Página de Agendamiento")  # Implementar según sea necesario
+
+def mostrar_pagina_principal():
+    st.title("VIRTUAL OMARDENT AI-BOTIDINAMIX")
+    st.markdown(
+        f"""
+        <style>
+        #video-container {{
+            position: relative;
+            width: 100%;
+            padding-bottom: 56.25%;
+            background-color: lightblue;
+            overflow: hidden;
+        }}
+        #background-video {{
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+        }}
+        </style>
+        <div id="video-container">
+            <video id="background-video" autoplay loop muted playsinline>
+                <source src="https://cdn.leonardo.ai/users/645c3d5c-ca1b-4ce8-aefa-a091494e0d09/generations/0c4f0fe7-5937-4644-b984-bdbd95018990/0c4f0fe7-5937-4644-b984-bdbd95018990.mp4" type="video/mp4">
+            </video>
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
+    archivo_pdf = st.file_uploader("📂 Cargar PDF", type='pdf', key='chat_pdf')
+
+    col1, col2 = st.columns([3, 1])
+    with col1:
+        pregunta_usuario = st.text_input("Pregunta:", key='unique_chat_input_key', value=st.session_state['transcripcion_voz'])
+    with col2:
+        capturar_voz()
+    
+    if pregunta_usuario:
+        manejar_pregunta_usuario(pregunta_usuario, archivo_pdf)
+
 def mostrar_galatea_asistente():
     st.markdown(
         """
@@ -529,15 +652,10 @@ def manejar_pregunta_usuario(pregunta_usuario, archivo_pdf=None):
     with st.chat_message("user"):
         st.markdown(pregunta_usuario)
 
-    texto_preprocesado = ""
-
+    texto_preprocesado = st.session_state.get('texto_pdf', "")
     if archivo_pdf:
         texto_pdf = extraer_texto_pdf(archivo_pdf)
         texto_preprocesado = preprocesar_texto(texto_pdf)
-    else:
-        with open(os.path.join("assets", "instrucciones.pdf"), "rb") as file:
-            texto_pdf = extraer_texto_pdf(file)
-            texto_preprocesado = preprocesar_texto(texto_pdf)
 
     # Obtener respuesta del modelo usando Assistant ID si está presente
     assistant_id = st.session_state.get('assistant_id', '')
@@ -662,125 +780,5 @@ def capturar_voz():
         unsafe_allow_html=True
     )
 
-def main():
-    st.set_page_config(page_title="Galatea OMARDENT", layout="wide")
-    
-    # Inicializar el estado de la sesión
-    if 'modelo' not in st.session_state:
-        st.session_state['modelo'] = "gpt-3.5-turbo"
-    if 'temperatura' not in st.session_state:
-        st.session_state['temperatura'] = 0.5
-    if 'mensajes_chat' not in st.session_state:
-        st.session_state['mensajes_chat'] = []
-    if 'transcripcion_voz' not in st.session_state:
-        st.session_state['transcripcion_voz'] = ""
-    if 'imagen_asistente' not in st.session_state:
-        st.session_state['imagen_asistente'] = None
-    if 'video_estado' not in st.session_state:
-        st.session_state['video_estado'] = 'paused'
-    if 'assistant_id' not in st.session_state:
-        st.session_state['assistant_id'] = 'asst_4ZYvBvf4IUVQPjnugSZGLdV2'
-    if 'presupuesto_texto' not in st.session_state:
-        st.session_state['presupuesto_texto'] = ''
-    if 'mostrar_chat' not in st.session_state:
-        st.session_state['mostrar_chat'] = False
-
-    # Barra lateral
-    ruta_logo = os.path.join("assets", "Logo Omardent.png")
-    if os.path.exists(ruta_logo):
-        st.sidebar.image(ruta_logo, use_column_width=True)
-    else:
-        st.sidebar.warning(f"Error: No se pudo encontrar la imagen en la ruta: {ruta_logo}")
-
-    st.sidebar.title("🤖 Galatea OMARDENT")
-    st.sidebar.markdown("---")
-    st.sidebar.subheader("🧠 Configuración del Modelo")
-    st.session_state['modelo'] = st.sidebar.selectbox(
-        "Selecciona el modelo:",
-        ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k", "gpt-4o"],
-        index=0,
-        key='modelo_selectbox',  # Clave única
-        help="Elige el modelo de lenguaje de OpenAI que prefieras."
-    )
-    st.sidebar.markdown("---")
-    st.session_state['temperatura'] = st.sidebar.slider(
-        "🌡️ Temperatura",
-        min_value=0.0, max_value=1.0,
-        value=st.session_state['temperatura'],
-        step=0.1,
-        key='temperatura_slider'  # Clave única
-    )
-    assistant_id = st.sidebar.text_input("Assistant ID", key="assistant_id", help="Introduce el Assistant ID del playground de OpenAI")
-    
-    st.sidebar.markdown("---")
-    st.sidebar.subheader("🌟 Navegación")
-    lateral_page = st.sidebar.radio("Ir a", ["Página Principal", "Gestión de Trabajos", "Gestión de Insumos", "Registro de Radiografías", "Buscar Datos", "Notificaciones", "Recomendaciones", "Asistente de Presupuestos", "Comunicación", "Asistente de Agendamiento"])
-
-    top_page = st.selectbox("Navegación Superior", ["Página Principal", "Galatea-Asistente"])
-
-    if top_page == "Galatea-Asistente":
-        mostrar_galatea_asistente()
-    else:
-        if lateral_page == "Página Principal":
-            mostrar_pagina_principal()
-        elif lateral_page == "Gestión de Trabajos":
-            flujo_laboratorio()
-        elif lateral_page == "Gestión de Insumos":
-            flujo_insumos()
-        elif lateral_page == "Registro de Radiografías":
-            flujo_radiografias()
-        elif lateral_page == "Buscar Datos":
-            buscar_datos_guardados()
-        elif lateral_page == "Notificaciones":
-            generar_notificaciones_pendientes()
-        elif lateral_page == "Recomendaciones":
-            mostrar_recomendaciones()
-        elif lateral_page == "Asistente de Presupuestos":
-            flujo_presupuestos()
-        elif lateral_page == "Comunicación":
-            st.write("Página de Comunicación")  # Implementar según sea necesario
-        elif lateral_page == "Asistente de Agendamiento":
-            st.write("Página de Agendamiento")  # Implementar según sea necesario
-
-def mostrar_pagina_principal():
-    st.title("VIRTUAL OMARDENT AI-BOTIDINAMIX")
-    st.markdown(
-        f"""
-        <style>
-        #video-container {{
-            position: relative;
-            width: 100%;
-            padding-bottom: 56.25%;
-            background-color: lightblue;
-            overflow: hidden;
-        }}
-        #background-video {{
-            position: absolute;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-        }}
-        </style>
-        <div id="video-container">
-            <video id="background-video" autoplay loop muted playsinline>
-                <source src="https://cdn.leonardo.ai/users/645c3d5c-ca1b-4ce8-aefa-a091494e0d09/generations/0c4f0fe7-5937-4644-b984-bdbd95018990/0c4f0fe7-5937-4644-b984-bdbd95018990.mp4" type="video/mp4">
-            </video>
-        </div>
-        """,
-        unsafe_allow_html=True
-    )
-
-    archivo_pdf = st.file_uploader("📂 Cargar PDF", type='pdf', key='chat_pdf')
-
-    col1, col2 = st.columns([3, 1])
-    with col1:
-        pregunta_usuario = st.text_input("Pregunta:", key='unique_chat_input_key', value=st.session_state['transcripcion_voz'])
-    with col2:
-        capturar_voz()
-    
-    if pregunta_usuario:
-        manejar_pregunta_usuario(pregunta_usuario, archivo_pdf)
-
 if __name__ == "__main__":
     main()