from pathlib import Path
from types import SimpleNamespace
import streamlit as st
import pandas as pd

from impuestos import TRAMOS, TRAMOS_REFORMA, get_table, get_curve

TOPES = SimpleNamespace(arriendo=450000, cuidado=550000)


def aplicar_beneficios(sueldo_bruto):
    "Referencias de: https://chocale.cl/2022/07/reforma-tributaria-gobierno-claves-proyecto-impuestos/"
    sb = st.sidebar
    sb.header("Beneficios tributarios")
    arriendo = sb.number_input(
        "Gastos de arriendo",
        value=450000,
        min_value=0,
        help="Se creará una exención que permitirá deducir de la base imponible del Impuesto Global Complementario los gastos de arriendo, con un tope de $450.000 mensuales",
    )
    # creditos = sb.checkbox("Tienes créditos?", value=False)
    cuidado = sb.number_input(
        "Tienes gastos asociados al cuidado de un familiar?",
        min_value=0,
        help="Se podrá deducir de la base imponible del Impuesto Global Complementario aquellos gastos que estén relacionados al cuidado de personas menores de dos años, o de personas con grados de dependencia severa. El tope será de $550.000 al mes.",
    )
    return sueldo_bruto - min(arriendo, TOPES.arriendo) - min(cuidado, TOPES.cuidado)


def main() -> None:
    st.header("Calcula tu impuesto a la renta :moneybag: :dollar: :bar_chart: con la Reforma Tributaria")

    with st.expander("Como se usa esta cosa?"):
        st.write(Path("info.md").read_text())

    sueldo_bruto = st.number_input(
        "Sueldo Bruto Mensual", value=1500000, min_value=300000
    )

    # calcular nueva base imponible
    sueldo_bruto_reforma = aplicar_beneficios(sueldo_bruto)

    st.text(f"Tu sueldo imponible antes de impuestos con la reforma es: {sueldo_bruto_reforma} (incluye los descuentos asociados a los beneficios)")


    

    col1, col2 = st.columns((0.5, 0.5))
    col1.subheader("Tabla de Impuestos Actual")
    table = get_table(sueldo_bruto, TRAMOS)
    col1.dataframe(table)
    total = table["Impuesto"].sum()
    col1.markdown(f"### Total Impuesto: \nActualmente pagas **${total}** que representa **({100*(total/sueldo_bruto):.2f}\%)** de tasa efectiva")

    col2.subheader("Tabla de Impuestos Reforma")
    table = get_table(sueldo_bruto_reforma, TRAMOS_REFORMA)
    col2.dataframe(table)
    total_reforma = table["Impuesto"].sum()
    explanation = f"### Total Impuesto con Reforma: \nPagarás **${total_reforma}** que representa **({100*(total_reforma/sueldo_bruto):.2f}\%)** de tasa efectiva"
    if total_reforma < total:
        explanation += " (Pagas menos que antes dado que estas beneficiando del descuento propocinado por los beneficios tributarios de arriendo y/o cuidado. Tu monto imponible es mas bajo que antes)"
    col2.markdown(explanation)

    # results_df = pd.DataFrame(columns=["actual", "reforma"], index=["total impuestos", "tasa efectiva"])
    # results_df["actual"] = [total, 100*(total/sueldo_bruto)]
    # results_df["reforma"] = [total_reforma, 100*(total_reforma/sueldo_bruto)]
    # st.dataframe(results_df.style.highlight_max(axis=1))
    
    st.markdown("### Objectivo de la reforma \nEn el siguiente grafico se ve claramente que la reforma empieza a tener efecto a partir de los 4 millones (pero muy lentamente 😱). Por ejemplo con 8 millones mensuales la diferencia es solo de 200000 mensual.")
    st.plotly_chart(get_curve(0), use_container_width=True)

if __name__ == "__main__":
    st.set_page_config(
        "Calculador de impuestos",
        "📊",
        initial_sidebar_state="expanded",
        layout="wide",
    )
    main()