Update app.py

app.py

@@ -4,33 +4,33 @@ import numpy as np
 import plotly.express as px
 
 def calculate_present_value(cash_flows, discount_rate):
-    years = np.arange(1, len(cash_flows)+1)
+    years = np.arange(1, len(cash_flows) + 1)
     present_values = cash_flows / (1 + discount_rate) ** years
     return np.sum(present_values)
 
-st.title('Calcula el Valor Presente')
+st.title('Calculadora de Valor Presente')
 
-# Input del usuario para cash flows
-st.subheader('Cash Flows x año:')
-cash_flows = []
-for i in range(1, 11):
-    cash_flow = st.number_input(f"Cash Flow {i}:", value=10000.0, step=1000.0, format="%.2f")
-    cash_flows.append(cash_flow)
+# Inicializar DataFrame para cash flows
+if 'cash_flows' not in st.session_state:
+    st.session_state.cash_flows = pd.DataFrame({
+        'Año': range(1, 11),
+        'Cash Flow': [10000.0] * 10  # Inicializar con un valor default
+    })
+
+# Mostrar la tabla editable
+st.subheader('Ingrese los Cash Flows anuales:')
+cash_flow_table = st.dataframe(st.session_state.cash_flows)
 
 # Input del usuario para la tasa de descuento
-discount_rate = st.number_input('Tasa de descuento (ejemplo: 0.05 para 5%):', value=0.05, step=0.01, format="%.2f")
+discount_rate = st.number_input('Ingrese la tasa de descuento (ejemplo: 0.05 para 5%):', value=0.05, step=0.01, format="%.2f")
 
+# Botón para calcular el valor presente
 if st.button('Calcular Valor Presente'):
-    cash_flows = np.array(cash_flows)
+    # Extraer los cash flows desde la tabla
+    cash_flows = st.session_state.cash_flows['Cash Flow'].values
     pv = calculate_present_value(cash_flows, discount_rate)
     st.success(f"El valor presente calculado es: ${pv:,.2f}")
 
-    # Crear DataFrame para visualización
-    df = pd.DataFrame({
-        'Año': np.arange(1, 11),
-        'Cash Flow': cash_flows
-    })
-
     # Gráfico
-    fig = px.scatter(df, x='Año', y='Cash Flow', title="Cash Flows vs. Tiempo", labels={'Cash Flow': 'Cash Flow ($)', 'Año': 'Año'})
+    fig = px.scatter(st.session_state.cash_flows, x='Año', y='Cash Flow', title="Cash Flows vs. Tiempo", labels={'Cash Flow': 'Cash Flow ($)', 'Año': 'Año'})
     st.plotly_chart(fig)