Update app.py

app.py

@@ -6,7 +6,6 @@ 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")
@@ -16,14 +15,14 @@ load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
 os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "botidinamix-g.json"
 
-# Función para obtener respuesta de OpenAI
-def obtener_respuesta(pregunta, modelo="gpt-4", temperatura=0.5):
+# Función para obtener respuesta de OpenAI con historial de conversación
+def obtener_respuesta(pregunta, historial, modelo="gpt-4", temperatura=0.5):
+    mensajes = historial + [
+        {"role": "user", "content": pregunta}
+    ]
     response = openai.ChatCompletion.create(
         model=modelo,
-        messages=[
-            {"role": "system", "content": "Eres Boty, un asistente de la empresa Botidinamix AI para el desarrollo e implementación de asistentes virtuales automatizados. Te guiarás a través de dos o tres preguntas para ayudarte a orientar y configurar tu asistente virtual. Actúas de manera muy amable, entusiasta y te diriges por el nombre a cada encuestado. Limita las preguntas adicionales a un máximo de 3 preguntas específicas."},
-            {"role": "user", "content": pregunta}
-        ],
+        messages=mensajes,
         temperature=temperatura,
         max_tokens=300,
     )
@@ -91,6 +90,7 @@ def encuesta_asistente():
 
     if 'respuestas' not in st.session_state:
         st.session_state.respuestas = {}
+        st.session_state.historial = [{"role": "system", "content": "Eres Boty, un asistente de la empresa Botidinamix AI para el desarrollo e implementación de asistentes virtuales automatizados. Te guiarás a través de dos o tres preguntas para ayudarte a orientar y configurar tu asistente virtual. Actúas de manera muy amable, entusiasta y te diriges por el nombre a cada encuestado. Limita las preguntas adicionales a un máximo de 3 preguntas específicas."}]
         st.session_state.pregunta_actual = 0
         st.session_state.preguntas_adicionales = []
         st.session_state.pregunta_adicional_actual = 0
@@ -102,7 +102,7 @@ def encuesta_asistente():
         st.write("Por favor, para poder continuar con el cuestionario es necesario que responda todas las preguntas adicionales.")
         pregunta_adicional_actual = st.session_state.pregunta_adicional_actual
         pregunta_adicional = st.session_state.preguntas_adicionales[pregunta_adicional_actual]
-        st.write(f"**Pregunta adicional {pregunta_adicional_actual + 1}:**")
+        st.write(f"**Pregunta adicional {pregunta_adicional_actual + 1}:** {pregunta_adicional}")
 
         st.session_state.respuestas_adicionales[f'respuesta_adicional_1_{pregunta_adicional_actual}'] = st.text_input("Respuesta adicional 1:", key=f"respuesta_adicional_1_{pregunta_adicional_actual}")
         st.session_state.respuestas_adicionales[f'respuesta_adicional_2_{pregunta_adicional_actual}'] = st.text_input("Respuesta adicional 2:", key=f"respuesta_adicional_2_{pregunta_adicional_actual}")
@@ -114,6 +114,11 @@ def encuesta_asistente():
                 st.session_state.respuestas[pregunta_adicional + " (Respuesta adicional 2)"] = st.session_state.respuestas_adicionales[f'respuesta_adicional_2_{pregunta_adicional_actual}']
                 st.session_state.respuestas[pregunta_adicional + " (Otras observaciones)"] = st.session_state.respuestas_adicionales[f'otras_observaciones_{pregunta_adicional_actual}']
 
+                st.session_state.historial.append({"role": "user", "content": st.session_state.respuestas_adicionales[f'respuesta_adicional_1_{pregunta_adicional_actual}']})
+                st.session_state.historial.append({"role": "user", "content": st.session_state.respuestas_adicionales[f'respuesta_adicional_2_{pregunta_adicional_actual}']})
+                st.session_state.historial.append({"role": "user", "content": st.session_state.respuestas_adicionales[f'otras_observaciones_{pregunta_adicional_actual}']})
+                st.session_state.historial.append({"role": "assistant", "content": st.session_state.respuesta_actual})
+
                 st.session_state.pregunta_adicional_actual += 1
 
                 if st.session_state.pregunta_adicional_actual == len(st.session_state.preguntas_adicionales):
@@ -139,11 +144,15 @@ def encuesta_asistente():
                     st.session_state.respuestas[pregunta_actual['pregunta']] = respuesta
 
                     # Obtener respuesta del asistente
-                    respuesta_asistente = obtener_respuesta(respuesta)
+                    respuesta_asistente = obtener_respuesta(respuesta, st.session_state.historial)
                     st.session_state.respuesta_actual = respuesta_asistente
 
+                    # Agregar la pregunta y respuesta actual al historial
+                    st.session_state.historial.append({"role": "user", "content": respuesta})
+                    st.session_state.historial.append({"role": "assistant", "content": respuesta_asistente})
+
                     # Verificar si el asistente hace una pregunta adicional
-                    if "pregunta" in respuesta_asistente.lower() and len(st.session_state.preguntas_adicionales) < 5:
+                    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
 
@@ -211,7 +220,7 @@ def incrustar_video_principal():
     video_bytes = open(video_path, 'rb').read()
     video_base64 = base64.b64encode(video_bytes).decode()
     video_html = f"""
-    <video width="100%" autoplay muted>
+    <video width="100%" autoplay>
         <source src="data:video/mp4;base64,{video_base64}" type="video/mp4">
         Tu navegador no soporta la reproducción de videos.
     </video>