import streamlit as st
import openai
import os
import tempfile
from google.cloud import texttospeech
from dotenv import load_dotenv
from docx import Document
import base64
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
import socket
import time

# Configuración de la interfaz
st.set_page_config(page_title="SURVEY ASSISTANT", layout="wide")

# Cargar variables de entorno
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 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=mensajes,
        temperature=temperatura,
        max_tokens=300,
    )
    respuesta = response['choices'][0]['message']['content']
    return respuesta

# Función para convertir texto a voz usando Google Cloud Text-to-Speech
def text_to_speech(text):
    client = texttospeech.TextToSpeechClient()
    synthesis_input = texttospeech.SynthesisInput(text=text)
    voice = texttospeech.VoiceSelectionParams(language_code="es-ES", ssml_gender=texttospeech.SsmlVoiceGender.NEUTRAL)
    audio_config = texttospeech.AudioConfig(audio_encoding=texttospeech.AudioEncoding.MP3)
    response = client.synthesize_speech(input=synthesis_input, voice=voice, audio_config=audio_config)
    audio_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3").name
    with open(audio_path, "wb") as out:
        out.write(response.audio_content)
    return audio_path

# Función para reproducir un archivo de audio
def reproducir_audio(file_path):
    audio_file = open(file_path, "rb")
    audio_data = audio_file.read()
    audio_base64 = base64.b64encode(audio_data).decode()
    audio_file.close()

    st.markdown(
        f"""
        <audio autoplay>
            <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
        </audio>
        """,
        unsafe_allow_html=True
    )

# Función para manejar la encuesta dinámica
def encuesta_asistente():
    if st.session_state.idioma == "Español":
        titulo = "Asistente Boti Encuesta"
        instrucciones = "Este asistente te guiará a través de una serie de preguntas para ayudarte a configurar tu asistente virtual"
        ayuda_texto = "Ayuda del asistente"
        preguntas = [
            {"pregunta": "¿Cuál es tu nombre?", "tipo": "texto"},
            {"pregunta": "¿Qué tipo de asistente virtual deseas?", "tipo": "multiple", "opciones": ["Asistente Personal", "Asistente de Negocios", "Asistente de Salud", "Asistente de Educación", "Otro"]},
            {"pregunta": "¿Qué objetivo debe cumplir este asistente?", "tipo": "multiple", "opciones": ["Organización Personal", "Mejora de Productividad", "Asistencia Médica", "Educación y Aprendizaje", "Automatizacion", "Otro"]},
            {"pregunta": "¿Cuáles son las funciones que debe tener?", "tipo": "multiple", "opciones": ["Gestión de Calendarios", "Recordatorios", "Consultas Médicas", "Ayuda con Tareas Escolares", "generacion creativa", "Otro"]},
            {"pregunta": "¿Cómo te gustaría que este asistente te ayude en tus tareas diarias?", "tipo": "multiple", "opciones": ["Planificación de Día", "Recordatorios", "Consultas y Respuestas", "Apoyo Emocional", "Otro"]},
            {"pregunta": "¿Qué interfaz debería tener este asistente?", "tipo": "multiple", "opciones": ["Aplicación Móvil", "Aplicación Web", "Asistente de Voz", "Chatbot en Redes Sociales", "Otro"]},
            {"pregunta": "¿Utilizarías el asistente para aumentar tu tiempo libre o para generar ingresos adicionales?", "tipo": "multiple", "opciones": ["Aumentar tiempo libre", "Generar ingresos adicionales"]},
            {"pregunta": "Por último, ¿qué estilos y elementos visuales debería tener?", "tipo": "multiple", "opciones": ["Estilo moderno con colores vivos", "Colores sobrios y clásico", "Moderno, colores pastel o simples", "Moderno, elegante, simple, no importa el estilo", "Otro"]}
        ]
        completar_encuesta = "Gracias por responder todas las preguntas. Haz clic en 'Registrar Encuesta' para finalizar y generar el archivo descargable."
        espere_mensaje = "Espere un minuto por favor mientras generamos su registro."
        mensaje_final = "¡No te vayas, tu asistente está casi listo!"
        encuesta_exitosa = "Encuesta completada exitosamente. Haz clic en 'Obtener Resultado de la Encuesta' para descargar el archivo."
        error_red = "No se pudo enviar el correo debido a un problema de red. Por favor, inténtelo más tarde."
        boton_descargar = "Descargar Archivo"
        informe_titulo = "Informe detallado del asistente solicitado"
    else:
        titulo = "Boti Survey Assistant"
        instrucciones = "This assistant will guide you through a series of questions to help you set up your virtual assistant"
        ayuda_texto = "Assistant Help"
        preguntas = [
            {"pregunta": "What is your name?", "tipo": "texto"},
            {"pregunta": "What type of virtual assistant do you want?", "tipo": "multiple", "opciones": ["Personal Assistant", "Business Assistant", "Health Assistant", "Education Assistant", "Other"]},
            {"pregunta": "What goal should this assistant achieve?", "tipo": "multiple", "opciones": ["Personal Organization", "Productivity Improvement", "Medical Assistance", "Education and Learning", "Automation", "Other"]},
            {"pregunta": "What functions should it have?", "tipo": "multiple", "opciones": ["Calendar Management", "Reminders", "Medical Consultations", "Help with School Tasks", "Creative Generation", "Other"]},
            {"pregunta": "How would you like this assistant to help you in your daily tasks?", "tipo": "multiple", "opciones": ["Day Planning", "Reminders", "Queries and Answers", "Emotional Support", "Other"]},
            {"pregunta": "What interface should this assistant have?", "tipo": "multiple", "opciones": ["Mobile Application", "Web Application", "Voice Assistant", "Chatbot on Social Networks", "Other"]},
            {"pregunta": "Would you use the assistant to increase your free time or to generate additional income?", "tipo": "multiple", "opciones": ["Increase free time", "Generate additional income"]},
            {"pregunta": "Lastly, what styles and visual elements should it have?", "tipo": "multiple", "opciones": ["Modern style with bright colors", "Sober and classic colors", "Modern, pastel or simple colors", "Modern, elegant, simple, no matter the style", "Other"]}
        ]
        completar_encuesta = "Thank you for answering all the questions. Click 'Register Survey' to finish and generate the downloadable file."
        espere_mensaje = "Please wait a minute while we generate your record."
        mensaje_final = "Don't go, your assistant is almost ready!"
        encuesta_exitosa = "Survey successfully completed. Click 'Get Survey Result' to download the file."
        error_red = "The email could not be sent due to a network problem. Please try again later."
        boton_descargar = "Download File"
        informe_titulo = "Detailed report of the requested assistant"

    st.title(titulo)
    st.write(instrucciones)

    # Logo de ayuda del asistente
    ayuda_logo = "https://img.freepik.com/premium-vector/minimal-ai-tech-robot-vector-illustration_589744-869.jpg"
    st.image(ayuda_logo, width=50)

    # Botón de ayuda del asistente
    if st.button(ayuda_texto):
        reproducir_audio("audios/instructivo.mp3")

    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
        st.session_state.respuestas_adicionales = {}
        st.session_state.respuestas_adicionales_completadas = False

    # Validación de preguntas adicionales
    if st.session_state.preguntas_adicionales and not st.session_state.respuestas_adicionales_completadas:
        st.write("Por favor, para poder continuar con el cuestionario es necesario que responda todas las preguntas adicionales." if st.session_state.idioma == "Español" else "Please, to continue with the survey you must answer all additional questions.")
        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_1 = st.text_input("Respuesta adicional 1:" if st.session_state.idioma == "Español" else "Additional answer 1:", key=f"respuesta_adicional_1_{pregunta_adicional_actual}")
            respuesta_adicional_2 = st.text_input("Respuesta adicional 2:" if st.session_state.idioma == "Español" else "Additional answer 2:", key=f"respuesta_adicional_2_{pregunta_adicional_actual}")
            otras_observaciones = st.text_input("Otras observaciones:" if st.session_state.idioma == "Español" else "Other observations:", key=f'otras_observaciones_{pregunta_adicional_actual}')
            
            if st.button("Enviar respuestas adicionales" if st.session_state.idioma == "Español" else "Send additional responses", key=f"btn_adicional_{pregunta_adicional_actual}"):
                with st.spinner('Procesando sus respuestas, por favor espere...' if st.session_state.idioma == "Español" else 'Processing your responses, please wait...'):
                    st.session_state.respuestas[pregunta_adicional + " (Respuesta adicional 1)"] = respuesta_adicional_1
                    st.session_state.respuestas[pregunta_adicional + " (Respuesta adicional 2)"] = respuesta_adicional_2
                    st.session_state.respuestas[pregunta_adicional + " (Otras observaciones)"] = otras_observaciones

                    st.session_state.historial.append({"role": "user", "content": respuesta_adicional_1})
                    st.session_state.historial.append({"role": "user", "content": respuesta_adicional_2})
                    st.session_state.historial.append({"role": "user", "content": otras_observaciones})
                    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):
                        st.session_state.respuestas_adicionales_completadas = True
                        st.session_state.preguntas_adicionales = []

                    st.experimental_rerun()
    else:
        if st.session_state.pregunta_actual < len(preguntas):
            pregunta_actual = preguntas[st.session_state.pregunta_actual]
            st.write(f"**Pregunta:** {pregunta_actual['pregunta']}")

            if pregunta_actual["tipo"] == "texto":
                respuesta = st.text_input("Tu respuesta:" if st.session_state.idioma == "Español" else "Your answer:", key=f"respuesta_{st.session_state.pregunta_actual}")
            elif pregunta_actual["tipo"] == "multiple":
                respuesta = st.selectbox("Selecciona una opción:" if st.session_state.idioma == "Español" else "Select an option:", pregunta_actual["opciones"], key=f"respuesta_{st.session_state.pregunta_actual}")
                if respuesta == "Otro" or respuesta == "Other":
                    respuesta_otro = st.text_input("Por favor especifica:" if st.session_state.idioma == "Español" else "Please specify:", key=f"respuesta_otro_{st.session_state.pregunta_actual}")
                    if respuesta_otro:
                        respuesta = respuesta_otro

            if st.button("Enviar respuesta" if st.session_state.idioma == "Español" else "Send answer", key=f"btn_{st.session_state.pregunta_actual}"):
                with st.spinner('Procesando su respuesta, por favor espere...' if st.session_state.idioma == "Español" else 'Processing your response, please wait...'):
                    st.session_state.respuestas[pregunta_actual['pregunta']] = respuesta

                    # Obtener respuesta del asistente
                    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) < 3:
                        st.session_state.preguntas_adicionales.append(respuesta_asistente)
                        st.session_state.respuestas_adicionales_completadas = False

                    # 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()
        else:
            # Mostrar el resumen de la encuesta antes de registrar
            st.write("NO OLVIDE DESCARGAR EL ARCHIVO AL FINALIZAR LA ENCUESTA" if st.session_state.idioma == "Español" else "DO NOT FORGET TO DOWNLOAD THE FILE AT THE END OF THE SURVEY")
            st.write("A continuación, se muestra un resumen del asistente de acuerdo a sus respuestas:" if st.session_state.idioma == "Español" else "Below is a summary of the assistant according to your answers:")
            resumen = generar_resumen(st.session_state.respuestas)
            st.write(resumen)
            
            st.write(completar_encuesta)
            if st.button("Registrar Encuesta" if st.session_state.idioma == "Español" else "Register Survey"):
                st.session_state.encuesta_completada = True
                st.markdown(f"<span style='color: white; font-weight: bold;'>{espere_mensaje}</span>", unsafe_allow_html=True)
                st.image("videos/robot-22.gif")
                time.sleep(60)  # Temporizador de 1 minuto
                st.markdown(f"<span style='color: yellow; font-size: 30px;'>{mensaje_final}</span>", unsafe_allow_html=True)
                try:
                    enviar_correo(resumen)
                    st.success(encuesta_exitosa)
                except (socket.gaierror, OSError):
                    st.error(error_red)
                generar_documento(resumen)

    if 'respuesta_actual' in st.session_state:
        st.write("Respuesta del asistente:" if st.session_state.idioma == "Español" else "Assistant's answer:")
        st.write(st.session_state.respuesta_actual)

        audio_path = st.session_state.audio_path
        audio_file = open(audio_path, "rb")
        audio_data = audio_file.read()
        audio_base64 = base64.b64encode(audio_data).decode()

        st.markdown(
            f"""
            <audio autoplay>
                <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
            </audio>
            """,
            unsafe_allow_html=True
        )

    if 'encuesta_completada' in st.session_state and st.session_state.encuesta_completada:
        generar_documento(resumen)

def generar_resumen(respuestas):
    # Generar un resumen en formato de texto
    resumen = "Este es el asistente configurado de acuerdo a sus respuestas y genera un resumen de ellas:\n\n" if st.session_state.idioma == "Español" else "This is the assistant configured according to your answers and generates a summary of them:\n\n"
    for pregunta, respuesta in respuestas.items():
        resumen += f"{pregunta}: {respuesta}\n"
    return resumen

def generar_documento(resumen):
    if 'respuestas' in st.session_state:
        respuestas = st.session_state.respuestas

        doc = Document()
        doc.add_heading('Asistente solicitado' if st.session_state.idioma == "Español" else 'Requested Assistant', 0)

        for pregunta, respuesta in respuestas.items():
            doc.add_heading(pregunta, level=1)
            doc.add_paragraph(respuesta)

        doc.add_page_break()
        doc.add_heading('Resumen del Asistente' if st.session_state.idioma == "Español" else 'Assistant Summary', level=1)
        doc.add_paragraph(resumen)

        informe = generar_informe_completo(respuestas)
        doc.add_heading('Informe Completo' if st.session_state.idioma == "Español" else 'Detailed Report', level=1)
        doc.add_paragraph(informe)

        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=boton_descargar,
                data=file,
                file_name="resultado_encuesta.docx",
                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
                key=f"btn_descargar_archivo_{temp_file.name}"
            )

def generar_informe_completo(respuestas):
    informe = "Informe detallado del asistente solicitado:\n\n" if st.session_state.idioma == "Español" else "Detailed report of the requested assistant:\n\n"
    # Generar un informe detallado basado en las respuestas del usuario
    # Asegurarse de que el informe tenga más de 1000 palabras
    for pregunta, respuesta in respuestas.items():
        informe += f"{pregunta}:\n"
        informe += f"{respuesta}\n\n"
    informe += "Este asistente virtual será diseñado para cubrir todas tus necesidades especificadas en la encuesta. Proporcionará soluciones personalizadas y funcionalidades avanzadas para mejorar tu productividad y bienestar. ¡Gracias por participar en esta encuesta!\n" if st.session_state.idioma == "Español" else "This virtual assistant will be designed to cover all your needs specified in the survey. It will provide personalized solutions and advanced functionalities to improve your productivity and well-being. Thank you for participating in this survey!\n"
    informe += "Es importante destacar que este asistente incluirá características específicas como la gestión de calendarios, recordatorios y apoyo emocional, entre otros. Su diseño será moderno y amigable, asegurando una experiencia de usuario óptima. Además, se adaptará a tus preferencias personales y objetivos, ya sea aumentar tu tiempo libre o generar ingresos adicionales.\n\n" if st.session_state.idioma == "Español" else "It is important to highlight that this assistant will include specific features such as calendar management, reminders, and emotional support, among others. Its design will be modern and user-friendly, ensuring an optimal user experience. Additionally, it will adapt to your personal preferences and goals, whether to increase your free time or generate additional income.\n\n"
    while len(informe.split()) < 1000:
        informe += "Este asistente será una herramienta invaluable para tu vida diaria, facilitando la organización, el aprendizaje y el manejo de tus responsabilidades. Con interfaces intuitivas y accesibles desde diversas plataformas, este asistente estará siempre disponible para ayudarte en cualquier momento y lugar. Su implementación no solo mejorará tu eficiencia sino también te brindará un soporte constante y personalizado.\n\n" if st.session_state.idioma == "Español" else "This assistant will be an invaluable tool for your daily life, facilitating organization, learning, and handling your responsibilities. With intuitive interfaces accessible from various platforms, this assistant will always be available to help you at any time and place. Its implementation will not only improve your efficiency but also provide you with constant and personalized support.\n\n"
    return informe

def enviar_correo(resumen):
    remitente = "tucorreo@gmail.com"
    destinatario = "josedcape@gmail.com"
    asunto = "Resumen de la Encuesta"
    cuerpo = resumen

    mensaje = MIMEMultipart()
    mensaje['From'] = remitente
    mensaje['To'] = destinatario
    mensaje['Subject'] = asunto

    mensaje.attach(MIMEText(cuerpo, 'plain'))

    # Iniciar la sesión SMTP
    servidor = smtplib.SMTP('smtp.gmail.com', 587)
    servidor.starttls()
    servidor.login(remitente, os.getenv("EMAIL_PASSWORD"))
    texto = mensaje.as_string()
    servidor.sendmail(remitente, destinatario, texto)
    servidor.quit()

# 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
    st_player(video_url, playing=True, volume=100)

# Estilos personalizados
st.markdown(
    """
    <style>
    .sidebar .sidebar-content {
        background-color: #2C3E50;
    }
    .sidebar .sidebar-content .element-container {
        color: #ECF0F1;
    }
    .main {
        background-color: #000000;
        color: #ECF0F1;
        background-image: url('https://images6.alphacoders.com/774/thumb-1920-774373.jpg');
        background-size: cover;
    }
    h1, h2, h3, h4, h5, h6 {
        color: #FFFFFF;
        font-weight: bold;
    }
    .big-button, .stButton button {
        font-size: 20px !important;
        padding: 10px !important;
        background-color: #007bff !important;
        color: white !important;
        border: none !important;
        text-align: center !important;
        cursor: pointer !important;
    }
    .stTextInput input {
        color: black !important;
        background-color: white !important;
    }
    .nav-bar {
        background-color: #000000;  /* Cambiado a negro */
        color: white;
        padding: 10px;
        text-align: center;
        font-size: 24px;
        font-weight: bold;
    }
    .nav-link {
        color: white;
        margin: 0 15px;
        text-decoration: none;
        cursor: pointer;
    }
    </style>
    """,
    unsafe_allow_html=True
)

# Función para la barra de navegación
def barra_navegacion():
    st.markdown(
        """
        <div class="nav-bar">
            <a class="nav-link" href="?page=principal">PAGINA PRINCIPAL</a>
            <a class="nav-link" href="?page=asistente">ASISTENTE ENCUESTA</a>
        </div>
        """,
        unsafe_allow_html=True
    )

# Función para seleccionar idioma
def seleccionar_idioma():
    idioma = st.sidebar.radio("Idioma", ("Español", "English"))
    st.session_state.idioma = idioma

# Funciones para las páginas
def pagina_principal():
    st.title("Bienvenido a Boti Asistente" if st.session_state.idioma == "Español" else "Welcome to Boti Assistant")
    st.write("Esta es la página principal de Boti Asistente, especializado en el desarrollo de bots para Botidinamix." if st.session_state.idioma == "Español" else "This is the main page of Boti Assistant, specialized in the development of bots for Botidinamix.")
    incrustar_video_principal()
    st.markdown("<h2 style='text-align: center; color: white;'>Para realizar su encuesta, haga clic en la barra superior</h2>" if st.session_state.idioma == "Español" else "<h2 style='text-align: center; color: white;'>To take your survey, click on the top bar</h2>", unsafe_allow_html=True)

def pagina_asistente():
    encuesta_asistente()

# Función principal
def main():
    if 'idioma' not in st.session_state:
        st.session_state.idioma = 'Español'
    
    seleccionar_idioma()
    barra_navegacion()
    
    query_params = st.experimental_get_query_params()
    page = query_params.get("page", ["principal"])[0]
    if page == "principal":
        pagina_principal()
    elif page == "asistente":
        pagina_asistente()

if __name__ == "__main__":
    main()