import streamlit as st
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

st.title('App de prueba')

A = st.number_input("Escribe la cantidad que ahorrarás ($): ")
T = st.number_input("Escribe cuantos meses quieres ver: ", step=1)

tasas = []
for i in range(int(T)):
    tasa = st.number_input(f"Tasa de interés nominal para el mes {i+1} (%): ")
    tasas.append(tasa)

if T > 0:
    z = np.arange(T)

    r_d = A * (1 + np.array(tasas) / (12 * 100)) ** z
    r_c = A * np.exp(np.array(tasas) * z / (12 * 100))

    new_dict = {"Mes": np.arange(1, T+1),
                "Retorno discreto": r_d,
                "Retorno continuo": r_c}

    df = pd.DataFrame(new_dict)
    st.dataframe(df)

    st.write(f"En {int(T)} meses tendrás: $ {r_c[-1]}")

    # Crear la gráfica
    plt.figure(figsize=(10, 6))
    plt.plot(np.arange(1, T+1), r_c, label='Retorno continuo', marker='o', linestyle='-')
    plt.xlabel('Mes')
    plt.ylabel('Inversión ($)')
    plt.title('Crecimiento de la inversión a lo largo de los meses')
    plt.grid(True)
    plt.legend()
    st.pyplot(plt)

else:
    st.write("Por favor, ingresa un valor de T mayor que 0.")