Update app.py

app.py

@@ -14,12 +14,13 @@ def calcular_cashflow(n, A, r, Pi):
 
 def graficar_cashflow(cashflow):
     meses = range(1, len(cashflow) + 1)
-    plt.plot(meses, cashflow, marker='o', linestyle='-')
-    plt.title('Cashflow por Mensualidad')
-    plt.xlabel('Mes')
-    plt.ylabel('Saldo')
-    plt.grid(True)
-    st.pyplot()
+    fig, ax = plt.subplots()
+    ax.plot(meses, cashflow, marker='o', linestyle='-')
+    ax.set_title('Cashflow por Mensualidad')
+    ax.set_xlabel('Mes')
+    ax.set_ylabel('Saldo')
+    ax.grid(True)
+    return fig, ax
 
 def main():
     st.title('Calculadora de Cashflow')
@@ -30,9 +31,8 @@ def main():
 
     if st.button('Calcular y Graficar Cashflow'):
         cashflow = calcular_cashflow(int(n), float(A), float(r), float(Pi))
-        graficar_cashflow(cashflow)
+        fig, ax = graficar_cashflow(cashflow)
+        st.pyplot(fig)
 
 if __name__ == "__main__":
     main()
-
-