#Importamos las librerías necesarias
import streamlit as st
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# La función que ayuda a realizar el valor presente 
def calculate_pv(cashflows, rate):
    pv = sum(cf / (1 + rate) ** i for i, cf in enumerate(cashflows))
    return pv

# Titulo de la página
st.title('Flujo de dinero: Calculadora de valor presente')

# Dar opción para meter la tasa de descuento
rate = st.number_input('Tasa de descuento (%)', min_value=0.0, max_value=100.0, value=10.0, step=0.1)/100

# Dar opción para meter cuantos años quiere calcular
n = st.number_input("Ingresa cuántos años quieres calcular", min_value = 0, step=1)

# Meter el flujo de dinero
cf_list = [0] * (n+1)
cf_dict = pd.DataFrame({"cashflows": cf_list}).astype(int)
cf_df = st.data_editor(cf_dict.T)

cashflows = np.array(cf_df.values[0])

# Genera el botón para calcular el valor presente
if st.button('Calcula valor presente'):
    pv = calculate_pv(cashflows, rate)
    st.write(f'El valor presente es: ${pv:,.2f}')
    
    # Para que se muestre en la gráfica los datos de valor presente
    colors = ['lightblue' if cf > 0 else 'lightsage' for cf in cashflows]
    plt.scatter(range(n+1), cashflows, c=colors)
    plt.title('Diagrama de flujo de dinero')
    plt.xlabel('Periodo')
    plt.ylabel('Flujo de dinero')
    for i, cf in enumerate(cashflows):
        plt.annotate(f'{cf:,.2f}', (i, cf), textcoords="offset points", xytext=(0,n), ha='center')
    st.pyplot(plt)