Update chat.py

chat.py

@@ -1,249 +1,249 @@
-import streamlit as st
-import openai
-from dotenv import load_dotenv
-import nltk
-import os
-import tempfile
-from nltk.tokenize import word_tokenize
-from nltk.corpus import stopwords
-from nltk.stem import SnowballStemmer
-import PyPDF2
-import time
-from google.cloud import texttospeech
-from Historial.historial_chat import cargar_historial, guardar_historial  # Importa las funciones
-from streamlit_webrtc import webrtc_streamer, WebRtcMode
-import av
-
-# Configuración de NLTK
-nltk.download('punkt')
-nltk.download('stopwords')
-
-# Función para cargar el texto del PDF
-def extraer_texto_pdf(archivo):
-    texto = ""
-    if archivo:
-        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
-            temp_file.write(archivo.read())
-            temp_file_path = temp_file.name
-        with open(temp_file_path, 'rb') as file:
-            reader = PyPDF2.PdfReader(file)
-            for page in range(len(reader.pages)):
-                texto += reader.pages[page].extract_text()
-        os.unlink(temp_file_path)
-    return texto
-
-# Función para preprocesar texto
-def preprocesar_texto(texto):
-    tokens = word_tokenize(texto, language='spanish')
-    tokens = [word.lower() for word in tokens if word.isalpha()]
-    stopwords_es = set(stopwords.words('spanish'))
-    tokens = [word for word in tokens if word not in stopwords_es]
-    stemmer = SnowballStemmer('spanish')
-    tokens = [stemmer.stem(word) for word in tokens]
-    return " ".join(tokens)
-
-# Cargar la clave API desde el archivo .env
-load_dotenv()
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "botidinamix-g.json"  # Reemplaza 'key.json' con el nombre de tu archivo de credenciales
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-# Instancia el cliente de Text-to-Speech
-client = texttospeech.TextToSpeechClient()
-
-# Función para obtener respuesta de OpenAI usando el modelo GPT y convertir a audio
-def obtener_respuesta(pregunta, texto_preprocesado, modelo, temperatura=0.5):
-    try:
-        response = openai.ChatCompletion.create(
-            model=modelo,
-            messages=[
-                {"role": "system", "content": "Actua como Elias un Teologo y psicologo capacitado, en la interpretacion de la palabra de Dios.   tienes un tono muy amable y cordial"},
-                {"role": "user", "content": f"{pregunta}\n\nContexto: {texto_preprocesado}"}
-            ],
-            temperature=temperatura
-        )
-        respuesta = response.choices[0].message['content'].strip()
-
-        # Configura la solicitud de síntesis de voz
-        input_text = texttospeech.SynthesisInput(text=respuesta)
-        voice = texttospeech.VoiceSelectionParams(
-            language_code="es-ES", ssml_gender=texttospeech.SsmlVoiceGender.MALE,
-        )
-        audio_config = texttospeech.AudioConfig(
-            audio_encoding=texttospeech.AudioEncoding.MP3
-        )
-
-        # Realiza la solicitud de síntesis de voz
-        response = client.synthesize_speech(
-            input=input_text, voice=voice, audio_config=audio_config
-        )
-
-        # Reproduce el audio en Streamlit
-        st.audio(response.audio_content, format="audio/mp3")
-        return respuesta
-
-    except openai.OpenAIError as e:
-        st.error(f"Error al comunicarse con OpenAI: {e}")
-        return "Lo siento, no puedo procesar tu solicitud en este momento."
-
-def main():
-    # --- Diseño general ---
-    st.set_page_config(page_title="Asistente Virtual", page_icon="🤖")
-
-    # --- Barra lateral ---
-    with st.sidebar:
-        st.image("Holybiblie.jpg")
-        st.title("🤖 LOS CODIGOS DE DIOS-BOTIDINAMIX AI")
-        st.markdown("---")
-          # --- Estilos CSS personalizados ---
-    st.markdown(
-        """
-    <style>
-        body {
-            background-image: linear-gradient(to right, #4B0082, #0000FF); /* Fondo degradado */
-        }
-        .stChatFloatingInputContainer { /* Estilos para el contenedor del input del chat */
-            background-color: rgba(255, 255, 255, 0.8); /* Fondo blanco con opacidad */
-            border-radius: 10px; /* Bordes redondeados */
-        }
-        .stTextInput > div > div > input {  /* Input de texto del chat */
-            color: #333; /* Color de texto más oscuro */
-        }
-        [data-testid="stChatMessage"] { /* Burbujas de chat */
-            background-color: black !important; /* Color de fondo negro */
-            color: gold !important; /* Color de texto dorado */
-            border-radius: 10px;
-        }
-        [data-testid="stChatMessage"] p { /* Párrafos dentro de las burbujas */
-            color: gold !important;
-        }
-        </style>
-        """,
-        unsafe_allow_html=True,
-    )
-        # --- Botones de historial ---
-    if st.button("Buscar Historial"):
-            st.session_state.mostrar_historial = True
-    if st.button("Borrar Historial"):
-            st.session_state.mensajes = []
-            st.session_state.mostrar_historial = False
-            st.success("Historial borrado correctamente")
-    # --- Chatbot ---
-    if 'mensajes' not in st.session_state:
-        st.session_state.mensajes = cargar_historial()
-
-    for mensaje in st.session_state.mensajes:
-        with st.chat_message(mensaje["role"]):
-            st.markdown(mensaje["content"])
-    # Función para manejar la entrada de audio
-    def on_audio(audio_bytes):
-        with st.spinner("Transcribiendo..."):
-            transcript = openai.Audio.transcribe("whisper-1", audio_bytes)
-            pregunta_usuario = transcript["text"]
-            st.session_state.mensajes.append({"role": "user", "content": pregunta_usuario, "timestamp": time.time()})
-            with st.chat_message("user"):
-                st.markdown(pregunta_usuario)
-
-    st.subheader("🎤 Captura de voz")
-    st.info("Haz clic en el micrófono y comienza a hablar. Tu pregunta se transcribirá automáticamente.")
-    with st.container():
-        if st.button("Grabar 🎙️"):
-            st.session_state.run_webrtc = True
-        if st.session_state.get("run_webrtc", False):
-            webrtc_streamer(
-                key="speech-to-text",
-                mode=WebRtcMode.SENDONLY,
-                rtc_configuration={"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]},
-                media_stream_constraints={"video": False, "audio": True},
-                on_audio=on_audio,
-            )
-            st.session_state["run_webrtc"] = False
-
-    # Mostrar historial si el botón fue presionado
-    if st.session_state.get("mostrar_historial", False):
-        st.subheader("Historial de Chat:")
-        if st.session_state.mensajes:
-            for mensaje in st.session_state.mensajes:
-                with st.chat_message(mensaje["role"]):
-                    st.markdown(mensaje["content"])
-        else:
-            st.info("No hay mensajes en el historial.")
-    else:
-        for mensaje in st.session_state.mensajes:
-            with st.chat_message(mensaje["role"]):
-                st.markdown(mensaje["content"])
-
-        # Selección de modelo de lenguaje
-        st.subheader("🧠 Configuración del Modelo")
-        modelo = st.selectbox(
-            "Selecciona el modelo:",
-            ["gpt-3.5-turbo", "gpt-4"],
-            index=0,
-            help="Elige el modelo de lenguaje de OpenAI que prefieras."
-        )
-
-        # --- Opciones adicionales ---
-        st.markdown("---")
-        temperatura = st.slider("🌡️ Temperatura", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
-
-        # --- Video de fondo ---
-        with st.container():
-            st.markdown(
-                f"""
-                <style>
-                #video-container {{
-                    position: relative;
-                    width: 100%;
-                    padding-bottom: 56.25%;
-                    background-color: lightblue;
-                    overflow: hidden;
-                }}
-                #background-video {{
-                    position: absolute;
-                    top: 0;
-                    left: 0;
-                    width: 100%;
-                    height: 100%;
-                }}
-                </style>
-                <div id="video-container">
-                    <video id="background-video" autoplay loop muted playsinline>
-                        <source src="https://cdn.pika.art/v1/f41adf48-a6db-4c6e-a93e-cd8db88f469d/movimiento_del_cuerpo_sutil_seed902109641145783.mp4" type="video/mp4">
-                    </video>
-                </div>
-                """,
-                unsafe_allow_html=True,
-            )
-
-        # --- Área principal de la aplicación ---
-        st.header("💬 Relexionar con Elias")
-
-        # Carga de archivo PDF
-        archivo_pdf = st.file_uploader("📂 Cargar PDF", type='pdf')
-
-        # --- Chatbot ---
-        if 'mensajes' not in st.session_state:
-            st.session_state.mensajes = []
-
-        for mensaje in st.session_state.mensajes:
-            with st.chat_message(mensaje["role"]):
-                st.markdown(mensaje["content"])
-
-        pregunta_usuario = st.chat_input("Pregunta:")
-        if pregunta_usuario:
-            st.session_state.mensajes.append({"role": "user", "content": pregunta_usuario})
-            with st.chat_message("user"):
-                st.markdown(pregunta_usuario)
-
-            if archivo_pdf:
-                texto_pdf = extraer_texto_pdf(archivo_pdf)
-                texto_preprocesado = preprocesar_texto(texto_pdf)
-            else:
-                texto_preprocesado = ""  # Sin contexto de PDF si no se carga un archivo
-
-            respuesta = obtener_respuesta(pregunta_usuario, texto_preprocesado, modelo, temperatura)
-            st.session_state.mensajes.append({"role": "assistant", "content": respuesta})
-            with st.chat_message("assistant"):
-                st.markdown(respuesta)
-
-if __name__ == "__main__":
-    main()
+import streamlit as st
+import openai
+from dotenv import load_dotenv
+import nltk
+import os
+import tempfile
+from nltk.tokenize import word_tokenize
+from nltk.corpus import stopwords
+from nltk.stem import SnowballStemmer
+import PyPDF2
+import time
+from google.cloud import texttospeech
+from Historial.historial_chat import cargar_historial, guardar_historial  # Importa las funciones
+from streamlit_webrtc import webrtc_streamer, WebRtcMode
+import av
+
+# Configuración de NLTK
+nltk.download('punkt')
+nltk.download('stopwords')
+
+# Función para cargar el texto del PDF
+def extraer_texto_pdf(archivo):
+    texto = ""
+    if archivo:
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            temp_file.write(archivo.read())
+            temp_file_path = temp_file.name
+        with open(temp_file_path, 'rb') as file:
+            reader = PyPDF2.PdfReader(file)
+            for page in range(len(reader.pages)):
+                texto += reader.pages[page].extract_text()
+        os.unlink(temp_file_path)
+    return texto
+
+# Función para preprocesar texto
+def preprocesar_texto(texto):
+    tokens = word_tokenize(texto, language='spanish')
+    tokens = [word.lower() for word in tokens if word.isalpha()]
+    stopwords_es = set(stopwords.words('spanish'))
+    tokens = [word for word in tokens if word not in stopwords_es]
+    stemmer = SnowballStemmer('spanish')
+    tokens = [stemmer.stem(word) for word in tokens]
+    return " ".join(tokens)
+
+# Cargar la clave API desde el archivo .env
+load_dotenv()
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "botidinamix-g.json"  # Reemplaza 'key.json' con el nombre de tu archivo de credenciales
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# Instancia el cliente de Text-to-Speech
+client = texttospeech.TextToSpeechClient()
+
+# Función para obtener respuesta de OpenAI usando el modelo GPT y convertir a audio
+def obtener_respuesta(pregunta, texto_preprocesado, modelo, temperatura=0.5):
+    try:
+        response = openai.ChatCompletion.create(
+            model=modelo,
+            messages=[
+                {"role": "system", "content": "Actua como Ana una asesora de ventas del restaurante Sazon Burguer, tienes un tono muy amable y cordial"},
+                {"role": "user", "content": f"{pregunta}\n\nContexto: {texto_preprocesado}"}
+            ],
+            temperature=temperatura
+        )
+        respuesta = response.choices[0].message['content'].strip()
+
+        # Configura la solicitud de síntesis de voz
+        input_text = texttospeech.SynthesisInput(text=respuesta)
+        voice = texttospeech.VoiceSelectionParams(
+            language_code="es-ES", ssml_gender=texttospeech.SsmlVoiceGender.MALE,
+        )
+        audio_config = texttospeech.AudioConfig(
+            audio_encoding=texttospeech.AudioEncoding.MP3
+        )
+
+        # Realiza la solicitud de síntesis de voz
+        response = client.synthesize_speech(
+            input=input_text, voice=voice, audio_config=audio_config
+        )
+
+        # Reproduce el audio en Streamlit
+        st.audio(response.audio_content, format="audio/mp3")
+        return respuesta
+
+    except openai.OpenAIError as e:
+        st.error(f"Error al comunicarse con OpenAI: {e}")
+        return "Lo siento, no puedo procesar tu solicitud en este momento."
+
+def main():
+    # --- Diseño general ---
+    st.set_page_config(page_title="Asistente Virtual", page_icon="🤖")
+
+    # --- Barra lateral ---
+    with st.sidebar:
+        st.image("Holybiblie.jpg")
+        st.title("🤖 LOS CODIGOS DE DIOS-BOTIDINAMIX AI")
+        st.markdown("---")
+          # --- Estilos CSS personalizados ---
+    st.markdown(
+        """
+    <style>
+        body {
+            background-image: linear-gradient(to right, #4B0082, #0000FF); /* Fondo degradado */
+        }
+        .stChatFloatingInputContainer { /* Estilos para el contenedor del input del chat */
+            background-color: rgba(255, 255, 255, 0.8); /* Fondo blanco con opacidad */
+            border-radius: 10px; /* Bordes redondeados */
+        }
+        .stTextInput > div > div > input {  /* Input de texto del chat */
+            color: #333; /* Color de texto más oscuro */
+        }
+        [data-testid="stChatMessage"] { /* Burbujas de chat */
+            background-color: black !important; /* Color de fondo negro */
+            color: gold !important; /* Color de texto dorado */
+            border-radius: 10px;
+        }
+        [data-testid="stChatMessage"] p { /* Párrafos dentro de las burbujas */
+            color: gold !important;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
+        # --- Botones de historial ---
+    if st.button("Buscar Historial"):
+            st.session_state.mostrar_historial = True
+    if st.button("Borrar Historial"):
+            st.session_state.mensajes = []
+            st.session_state.mostrar_historial = False
+            st.success("Historial borrado correctamente")
+    # --- Chatbot ---
+    if 'mensajes' not in st.session_state:
+        st.session_state.mensajes = cargar_historial()
+
+    for mensaje in st.session_state.mensajes:
+        with st.chat_message(mensaje["role"]):
+            st.markdown(mensaje["content"])
+    # Función para manejar la entrada de audio
+    def on_audio(audio_bytes):
+        with st.spinner("Transcribiendo..."):
+            transcript = openai.Audio.transcribe("whisper-1", audio_bytes)
+            pregunta_usuario = transcript["text"]
+            st.session_state.mensajes.append({"role": "user", "content": pregunta_usuario, "timestamp": time.time()})
+            with st.chat_message("user"):
+                st.markdown(pregunta_usuario)
+
+    st.subheader("🎤 Captura de voz")
+    st.info("Haz clic en el micrófono y comienza a hablar. Tu pregunta se transcribirá automáticamente.")
+    with st.container():
+        if st.button("Grabar 🎙️"):
+            st.session_state.run_webrtc = True
+        if st.session_state.get("run_webrtc", False):
+            webrtc_streamer(
+                key="speech-to-text",
+                mode=WebRtcMode.SENDONLY,
+                rtc_configuration={"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]},
+                media_stream_constraints={"video": False, "audio": True},
+                on_audio=on_audio,
+            )
+            st.session_state["run_webrtc"] = False
+
+    # Mostrar historial si el botón fue presionado
+    if st.session_state.get("mostrar_historial", False):
+        st.subheader("Historial de Chat:")
+        if st.session_state.mensajes:
+            for mensaje in st.session_state.mensajes:
+                with st.chat_message(mensaje["role"]):
+                    st.markdown(mensaje["content"])
+        else:
+            st.info("No hay mensajes en el historial.")
+    else:
+        for mensaje in st.session_state.mensajes:
+            with st.chat_message(mensaje["role"]):
+                st.markdown(mensaje["content"])
+
+        # Selección de modelo de lenguaje
+        st.subheader("🧠 Configuración del Modelo")
+        modelo = st.selectbox(
+            "Selecciona el modelo:",
+            ["gpt-3.5-turbo", "gpt-4"],
+            index=0,
+            help="Elige el modelo de lenguaje de OpenAI que prefieras."
+        )
+
+        # --- Opciones adicionales ---
+        st.markdown("---")
+        temperatura = st.slider("🌡️ Temperatura", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+
+        # --- Video de fondo ---
+        with st.container():
+            st.markdown(
+                f"""
+                <style>
+                #video-container {{
+                    position: relative;
+                    width: 100%;
+                    padding-bottom: 56.25%;
+                    background-color: lightblue;
+                    overflow: hidden;
+                }}
+                #background-video {{
+                    position: absolute;
+                    top: 0;
+                    left: 0;
+                    width: 100%;
+                    height: 100%;
+                }}
+                </style>
+                <div id="video-container">
+                    <video id="background-video" autoplay loop muted playsinline>
+                        <source src="https://cdn.pika.art/v1/f41adf48-a6db-4c6e-a93e-cd8db88f469d/movimiento_del_cuerpo_sutil_seed902109641145783.mp4" type="video/mp4">
+                    </video>
+                </div>
+                """,
+                unsafe_allow_html=True,
+            )
+
+        # --- Área principal de la aplicación ---
+        st.header("💬 Relexionar con Elias")
+
+        # Carga de archivo PDF
+        archivo_pdf = st.file_uploader("📂 Cargar PDF", type='pdf')
+
+        # --- Chatbot ---
+        if 'mensajes' not in st.session_state:
+            st.session_state.mensajes = []
+
+        for mensaje in st.session_state.mensajes:
+            with st.chat_message(mensaje["role"]):
+                st.markdown(mensaje["content"])
+
+        pregunta_usuario = st.chat_input("Pregunta:")
+        if pregunta_usuario:
+            st.session_state.mensajes.append({"role": "user", "content": pregunta_usuario})
+            with st.chat_message("user"):
+                st.markdown(pregunta_usuario)
+
+            if archivo_pdf:
+                texto_pdf = extraer_texto_pdf(archivo_pdf)
+                texto_preprocesado = preprocesar_texto(texto_pdf)
+            else:
+                texto_preprocesado = ""  # Sin contexto de PDF si no se carga un archivo
+
+            respuesta = obtener_respuesta(pregunta_usuario, texto_preprocesado, modelo, temperatura)
+            st.session_state.mensajes.append({"role": "assistant", "content": respuesta})
+            with st.chat_message("assistant"):
+                st.markdown(respuesta)
+
+if __name__ == "__main__":
+    main()