Update app.py

app.py

@@ -205,12 +205,13 @@ def encuesta_asistente():
             if st.button("Registrar Encuesta"):
                 st.session_state.encuesta_completada = True
                 st.markdown("<span style='color: white; font-weight: bold;'>Espere un minuto por favor mientras generamos su registro.</span>", unsafe_allow_html=True)
+                st.image("videos/robot-22.gif")
                 try:
                     enviar_correo(resumen)
                     st.success("Encuesta completada exitosamente. Haz clic en 'Obtener Resultado de la Encuesta' para descargar el archivo.")
                 except (socket.gaierror, OSError):
                     st.error("No se pudo enviar el correo debido a un problema de red. Por favor, inténtelo más tarde.")
-                generar_documento()
+                generar_documento(resumen)
 
     if 'respuesta_actual' in st.session_state:
         st.write("Respuesta del asistente:")
@@ -231,7 +232,7 @@ def encuesta_asistente():
         )
 
     if 'encuesta_completada' in st.session_state and st.session_state.encuesta_completada:
-        generar_documento()
+        generar_documento(resumen)
 
 def generar_resumen(respuestas):
     # Generar un resumen en formato de texto
@@ -240,17 +241,20 @@ def generar_resumen(respuestas):
         resumen += f"{pregunta}: {respuesta}\n"
     return resumen
 
-def generar_documento():
+def generar_documento(resumen):
     if 'respuestas' in st.session_state:
         respuestas = st.session_state.respuestas
 
         doc = Document()
-        doc.add_heading('Respuestas de la Encuesta y resumen', 0)
+        doc.add_heading('Respuestas de la Encuesta y Resumen', 0)
 
         for pregunta, respuesta in respuestas.items():
             doc.add_heading(pregunta, level=1)
             doc.add_paragraph(respuesta)
 
+        doc.add_heading('Resumen del Asistente', level=1)
+        doc.add_paragraph(resumen)
+
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".docx")
         doc.save(temp_file.name)