import streamlit as st


def Contact():
    st.markdown('''
                    <h2>Simulación de <strong>Presupuesto</strong></h2>
                    <hr>
                ''', unsafe_allow_html=True)

    # Define los precios para cada opción
    precio_analisis = 3500
    precio_ML = 5500
    precio_app = 3000
    precio_mantenimiento = 550

    # Define las opciones como un diccionario de la forma {nombre_opción: precio_opción}
    opciones = {
        "Opción 1": precio_analisis,
        "Opción 2": precio_ML,
        "Opción 3": precio_app,
        "Opción 4": precio_mantenimiento
    }

    # Crea un checkbox para cada opción
    analisis = st.checkbox("Análisis y Reportes")
    ML = st.checkbox("Algoritmos de ML aplicado")
    app = st.checkbox("Creación de una App")
    # mantenimiento = st.checkbox('Mantenimiento')

    # Crea un campo numérico para la cantidad
    cantidad = st.number_input("Meses de mantenimiento:", min_value=0, value=0)

    mes_mantenimiento = cantidad * precio_mantenimiento

    # Calcula el total en función de las opciones elegidas
    total = mes_mantenimiento + sum(
        [opciones[opcion] for opcion, seleccionada in zip(opciones.keys(),
                                                          [analisis, ML,
                                                           app]) if
         seleccionada])

    # Muestra el total
    st.markdown(f'''
            Total $: <span style="background:#5c62ac;\
            border-radius:5px;padding:5px">{total}</span> 
            :rocket:
        ''', unsafe_allow_html=True)

    st.markdown('''
                <br>
                <h2>Datos de contactos</h2>
                <hr>
            ''', unsafe_allow_html=True)
    # Crea campos de entrada para el nombre, correo electrónico y mensaje
    nombre = st.text_input("Nombre completo")
    email = st.text_input("Correo electrónico")
    mensaje = st.text_area("Mensaje")

    # Crea un botón para enviar el formulario
    enviar = st.button("Enviar")

    # Si el botón es presionado, muestra un mensaje de confirmación
    if enviar:
        if nombre and email and mensaje:

            st.write(
                "¡Gracias por tu mensaje! Nos pondremos en contacto contigo pronto.")
        else:
            st.error("Por favor completa todos los campos requeridos.")