Update app.py

app.py

@@ -12,8 +12,6 @@ from fpdf import FPDF
 import streamlit as st
 import requests
 from google.cloud import texttospeech
-import io
-from docx import Document
 
 nltk.download('punkt', quiet=True)
 nltk.download('stopwords', quiet=True)
@@ -302,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,
@@ -459,28 +457,92 @@ def mostrar_recomendaciones():
     st.title("⭐ Recomendaciones")
     st.write("Aquí puedes encontrar recomendaciones y consejos útiles.")
 
+def mostrar_galatea_asistente():
+    st.markdown(
+        """
+        <style>
+        #video-container {
+            position: relative;
+            width: 100%;
+            height: 40vh;
+            background-color: lightblue;
+            overflow: hidden;
+            display: flex;
+            justify-content: center;
+            align-items: center;
+        }
+        #background-video {
+            width: 100%;
+            height: auto;
+        }
+        #chat-container {
+            margin-top: 20px;
+            width: 100%;
+            display: flex;
+            flex-direction: column;
+            justify-content: center;
+            align-items: center;
+        }
+        .chat-message {
+            background: rgba(255, 255, 255, 0.8);
+            border-radius: 10px;
+            padding: 10px;
+            margin-bottom: 10px;
+            max-width: 70%;
+        }
+        .chat-message.user {
+            align-self: flex-start;
+        }
+        .chat-message.assistant {
+            align-self: flex-end;
+            background: rgba(0, 123, 255, 0.8);
+            color: white;
+        }
+        </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/aaa569a1-8952-4e8e-9c3c-71cc66e62f04/aaa569a1-8952-4e8e-9c3c-71cc66e62f04.mp4" type="video/mp4">
+            </video>
+        </div>
+        <div id="chat-container">
+        """,
+        unsafe_allow_html=True
+    )
+
+    for mensaje in st.session_state['mensajes_chat']:
+        clase = "user" if mensaje["role"] == "user" else "assistant"
+        st.markdown(f'<div class="chat-message {clase}">{mensaje["content"]}</div>', unsafe_allow_html=True)
+
+    pregunta_usuario = st.text_input("Escribe tu pregunta aquí:", key='unique_chat_input_key', value=st.session_state['transcripcion_voz'])
+    if st.button("Enviar Pregunta"):
+        manejar_pregunta_usuario(pregunta_usuario)
+
+    st.markdown("</div>", unsafe_allow_html=True)
+
+    if st.session_state['imagen_asistente']:
+        st.image(st.session_state['imagen_asistente'], use_column_width=True)
+    else:
+        st.warning("No se ha cargado ninguna imagen. Por favor, carga una imagen en la página principal.")
+
 def manejar_pregunta_usuario(pregunta_usuario, archivo_pdf=None):
     st.session_state['mensajes_chat'].append({"role": "user", "content": pregunta_usuario})
     with st.chat_message("user"):
         st.markdown(pregunta_usuario)
 
     texto_preprocesado = ""
+
     if archivo_pdf:
         texto_pdf = extraer_texto_pdf(archivo_pdf)
         texto_preprocesado = preprocesar_texto(texto_pdf)
-
-    # Cargar y procesar el documento "Instrucciones Galatea.pdf"
-    with open("assets","instrucciones.pdf", "rb") as file:
-        instrucciones_pdf = extraer_texto_pdf(file)
-        instrucciones_preprocesado = preprocesar_texto(instrucciones_pdf)
-    
-    # Combinar el texto preprocesado del PDF cargado con las instrucciones predefinidas
-    contexto = instrucciones_preprocesado + "\n" + texto_preprocesado
+    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', '')
     if assistant_id:
-        prompt = f"{contexto}\n\n{pregunta_usuario}"
+        prompt = f"{texto_preprocesado}\n\n{pregunta_usuario}"
         response = openai.ChatCompletion.create(
             model=st.session_state['modelo'],
             messages=[
@@ -494,7 +556,7 @@ def manejar_pregunta_usuario(pregunta_usuario, archivo_pdf=None):
     else:
         respuesta = obtener_respuesta(
             pregunta_usuario,
-            contexto,
+            texto_preprocesado,
             st.session_state['modelo'],
             st.session_state['temperatura'],
             assistant_id
@@ -720,72 +782,5 @@ def mostrar_pagina_principal():
     if pregunta_usuario:
         manejar_pregunta_usuario(pregunta_usuario, archivo_pdf)
 
-def mostrar_galatea_asistente():
-    st.markdown(
-        """
-        <style>
-        #video-container {
-            position: relative;
-            width: 100%;
-            height: 40vh;
-            background-color: lightblue;
-            overflow: hidden;
-            display: flex;
-            justify-content: center;
-            align-items: center;
-        }
-        #background-video {
-            width: 100%;
-            height: auto;
-        }
-        #chat-container {
-            margin-top: 20px;
-            width: 100%;
-            display: flex;
-            flex-direction: column;
-            justify-content: center;
-            align-items: center;
-        }
-        .chat-message {
-            background: rgba(255, 255, 255, 0.8);
-            border-radius: 10px;
-            padding: 10px;
-            margin-bottom: 10px;
-            max-width: 70%;
-        }
-        .chat-message.user {
-            align-self: flex-start;
-        }
-        .chat-message.assistant {
-            align-self: flex-end;
-            background: rgba(0, 123, 255, 0.8);
-            color: white;
-        }
-        </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/aaa569a1-8952-4e8e-9c3c-71cc66e62f04/aaa569a1-8952-4e8e-9c3c-71cc66e62f04.mp4" type="video/mp4">
-            </video>
-        </div>
-        <div id="chat-container">
-        """,
-        unsafe_allow_html=True
-    )
-
-    for mensaje in st.session_state['mensajes_chat']:
-        clase = "user" if mensaje["role"] == "user" else "assistant"
-        st.markdown(f'<div class="chat-message {clase}">{mensaje["content"]}</div>', unsafe_allow_html=True)
-
-    pregunta_usuario = st.text_input("Escribe tu pregunta aquí:", key='unique_chat_input_key', value=st.session_state['transcripcion_voz'])
-    if st.button("Enviar Pregunta"):
-        manejar_pregunta_usuario(pregunta_usuario)
-
-    st.markdown("</div>", unsafe_allow_html=True)
-
-    if st.session_state['imagen_asistente']:
-        st.image(st.session_state['imagen_asistente'], use_column_width=True)
-    else:
-        st.warning("No se ha cargado ninguna imagen. Por favor, carga una imagen en la página principal.")
-
 if __name__ == "__main__":
     main()