Update app.py

app.py

@@ -6,6 +6,7 @@ from google.cloud import texttospeech
 from dotenv import load_dotenv
 from docx import Document
 import base64
+import time
 
 # Configuración de la interfaz
 st.set_page_config(page_title="Boti Asistente", layout="wide")
@@ -92,6 +93,7 @@ def encuesta_asistente():
         st.session_state.respuestas = {}
         st.session_state.pregunta_actual = 0
         st.session_state.preguntas_adicionales = []
+        st.session_state.pregunta_adicional_actual = 0
 
     if st.session_state.pregunta_actual < len(preguntas):
         pregunta_actual = preguntas[st.session_state.pregunta_actual]
@@ -107,22 +109,23 @@ def encuesta_asistente():
                     respuesta = respuesta_otro
         
         if st.button("Enviar respuesta", key=f"btn_{st.session_state.pregunta_actual}"):
-            st.session_state.respuestas[pregunta_actual['pregunta']] = respuesta
+            with st.spinner('Procesando su respuesta, por favor espere...'):
+                st.session_state.respuestas[pregunta_actual['pregunta']] = respuesta
 
-            # Obtener respuesta del asistente
-            respuesta_asistente = obtener_respuesta(respuesta)
-            st.session_state.respuesta_actual = respuesta_asistente
+                # Obtener respuesta del asistente
+                respuesta_asistente = obtener_respuesta(respuesta)
+                st.session_state.respuesta_actual = respuesta_asistente
 
-            # Verificar si el asistente hace una pregunta adicional
-            if "pregunta" in respuesta_asistente.lower() and len(st.session_state.preguntas_adicionales) < 3:
-                st.session_state.preguntas_adicionales.append(respuesta_asistente)
+                # Verificar si el asistente hace una pregunta adicional
+                if "pregunta" in respuesta_asistente.lower() and len(st.session_state.preguntas_adicionales) < 3:
+                    st.session_state.preguntas_adicionales.append(respuesta_asistente)
 
-            # Convertir respuesta a audio
-            audio_path = text_to_speech(respuesta_asistente)
-            st.session_state.audio_path = audio_path
+                # Convertir respuesta a audio
+                audio_path = text_to_speech(respuesta_asistente)
+                st.session_state.audio_path = audio_path
 
-            st.session_state.pregunta_actual += 1
-            st.experimental_rerun()
+                st.session_state.pregunta_actual += 1
+                st.experimental_rerun()
     else:
         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"):
@@ -150,11 +153,13 @@ def encuesta_asistente():
     # Preguntas adicionales
     if st.session_state.preguntas_adicionales:
         st.markdown("<h3>Preguntas Adicionales:</h3>", unsafe_allow_html=True)
-        for i, pregunta_adicional in enumerate(st.session_state.preguntas_adicionales):
-            with st.expander(f"Pregunta adicional {i+1}"):
-                st.write(f"**Pregunta:** {pregunta_adicional}")
-                respuesta_adicional = st.text_input("Tu respuesta:", key=f"respuesta_adicional_{i}")
-                if st.button("Enviar respuesta adicional", key=f"btn_adicional_{i}"):
+        pregunta_adicional_actual = st.session_state.pregunta_adicional_actual
+        if pregunta_adicional_actual < len(st.session_state.preguntas_adicionales):
+            pregunta_adicional = st.session_state.preguntas_adicionales[pregunta_adicional_actual]
+            st.write(f"**Pregunta adicional {pregunta_adicional_actual + 1}:** {pregunta_adicional}")
+            respuesta_adicional = st.text_input("Tu respuesta:", key=f"respuesta_adicional_{pregunta_adicional_actual}")
+            if st.button("Enviar respuesta adicional", key=f"btn_adicional_{pregunta_adicional_actual}"):
+                with st.spinner('Procesando su respuesta, por favor espere...'):
                     st.session_state.respuestas[pregunta_adicional] = respuesta_adicional
                     respuesta_asistente = obtener_respuesta(respuesta_adicional)
                     st.session_state.respuesta_actual = respuesta_asistente
@@ -162,6 +167,7 @@ def encuesta_asistente():
                     audio_path = text_to_speech(respuesta_asistente)
                     st.session_state.audio_path = audio_path
 
+                    st.session_state.pregunta_adicional_actual += 1
                     st.experimental_rerun()
 
     if 'encuesta_completada' in st.session_state and st.session_state.encuesta_completada: