Update app.py

app.py

@@ -8,15 +8,17 @@ def calculate_returns(y_initial, A, T, changes_in_interest_rate):
     z = np.arange(T+1)
     r_c = np.zeros(T+1)
     interest_rate = y_initial / (12 * 100)
+    interest_rates = [interest_rate] * (T+1)
     for month in range(T+1):
         for change_month, new_interest_rate in changes_in_interest_rate.items():
             if month >= change_month:
                 interest_rate = new_interest_rate / (12 * 100)
+                interest_rates[month] = interest_rate
         r_c[month] = A * np.exp(interest_rate * month)
-    return z, r_c
+    return z, r_c, interest_rates
 
 def main():
-    st.title('App de pruebas')
+    st.title('App de prueba 🏦')
 
     # Ingresar la tasa de interés nominal inicial
     y_initial = st.number_input("Escribe la tasa de interés nominal inicial (%):")
@@ -38,11 +40,11 @@ def main():
         new_interest_rate = st.number_input(f"Nueva tasa de interés a partir del mes {change_month} (%):")
         changes_in_interest_rate[change_month] = new_interest_rate
 
-    # Calcular los retornos
-    z, r_c = calculate_returns(y_initial, A, T, changes_in_interest_rate)
+    # Calcular los retornos y las tasas de interés correspondientes
+    z, r_c, interest_rates = calculate_returns(y_initial, A, T, changes_in_interest_rate)
 
     # Crear un DataFrame con los resultados
-    new_dict = {"Mes": z, "Retorno continuo": r_c}
+    new_dict = {"Mes": z, "Tasa de Interés (%)": interest_rates, "Retorno continuo": r_c}
     df = pd.DataFrame(new_dict)
 
     # Mostrar el DataFrame y el resultado final
@@ -53,8 +55,4 @@ def main():
     plt.plot(z, r_c)
     plt.xlabel('Mes')
     plt.ylabel('Retorno')
-    plt.title('Comportamiento de la inversión')
-    st.pyplot()
-
-if __name__ == "__main__":
-    main()
+    plt.