Update app.py

app.py

@@ -10,6 +10,7 @@ from streamlit_player import st_player
 import smtplib
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
+from email.mime.application import MIMEApplication
 
 # Configuración de la interfaz
 st.set_page_config(page_title="SURVEY ASSISTANT", layout="wide")
@@ -182,7 +183,7 @@ def encuesta_asistente():
                     st.session_state.historial.append({"role": "assistant", "content": respuesta_asistente})
 
                     # Verificar si el asistente hace una pregunta adicional
-                    if "pregunta" en respuesta_asistente.lower() and len(st.session_state.preguntas_adicionales) < 3:
+                    if "pregunta" in respuesta_asistente.lower() and len(st.session_state.preguntas_adicionales) < 3:
                         st.session_state.preguntas_adicionales.append(respuesta_asistente)
                         st.session_state.respuestas_adicionales_completadas = False
 
@@ -202,7 +203,6 @@ def encuesta_asistente():
             st.write("Gracias por responder todas las preguntas. Haz clic en 'Registrar Encuesta' para finalizar y generar el archivo descargable.")
             if st.button("Registrar Encuesta"):
                 st.session_state.encuesta_completada = True
-                generar_documento()  # Generar el documento al finalizar la encuesta
                 enviar_correo(resumen)
                 st.success("Encuesta completada exitosamente. Haz clic en 'Obtener Resultado de la Encuesta' para descargar el archivo.")
 
@@ -241,14 +241,29 @@ def encuesta_asistente():
             """,
             unsafe_allow_html=True
         )
-        # Mostrar el botón de descarga del documento
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".docx")
+        generar_documento()
+
+def generar_resumen(respuestas):
+    # Generar un resumen en formato de texto
+    resumen = "Este es el asistente configurado de acuerdo a sus respuestas:\n\n"
+    for pregunta, respuesta in respuestas.items():
+        resumen += f"{pregunta}: {respuesta}\n"
+    return resumen
+
+def generar_documento():
+    if 'respuestas' in st.session_state:
+        respuestas = st.session_state.respuestas
+
         doc = Document()
         doc.add_heading('Respuestas de la Encuesta', 0)
-        for pregunta, respuesta in st.session_state.respuestas.items():
+
+        for pregunta, respuesta in respuestas.items():
             doc.add_heading(pregunta, level=1)
             doc.add_paragraph(respuesta)
+
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".docx")
         doc.save(temp_file.name)
+
         with open(temp_file.name, "rb") as file:
             st.download_button(
                 label="Descargar Archivo",
@@ -258,13 +273,6 @@ def encuesta_asistente():
                 key="btn_descargar_archivo"
             )
 
-def generar_resumen(respuestas):
-    # Generar un resumen en formato de texto
-    resumen = "Este es el asistente configurado de acuerdo a sus respuestas:\n\n"
-    for pregunta, respuesta in respuestas.items():
-        resumen += f"{pregunta}: {respuesta}\n"
-    return resumen
-
 def enviar_correo(resumen):
     remitente = "tucorreo@gmail.com"
     destinatario = "josedcape@gmail.com"
@@ -288,7 +296,7 @@ def enviar_correo(resumen):
 
 # Función para incrustar video en la página principal
 def incrustar_video_principal():
-    video_url = "https://www.youtube.com/watch?v=uGzZe1LxVPk"  # Reemplaza con el enlace correcto
+    video_url = "https://www.youtube.com/watch?v=uGzZe1LxVPk"
     st_player(video_url, playing=True, volume=100)
 
 # Estilos personalizados
@@ -391,14 +399,5 @@ def main():
     elif page == "video":
         pagina_ver_video()
 
-    # Botón para deshacer cambios y volver al menú principal
-    if st.sidebar.button("Deshacer cambios"):
-        st.session_state.clear()
-        st.experimental_rerun()
-
-    if st.sidebar.button("Volver al menú principal"):
-        st.experimental_set_query_params(page="principal")
-        st.experimental_rerun()
-
 if __name__ == "__main__":
     main()