Update app.py

app.py

@@ -185,21 +185,21 @@ def get_array_CF_PV():
         
         profit = get_profit(array_amortization[idx], idx)
         
-        array_CF.append(profit)
+        array_CF.append(round(profit))
         
         cur_PV = get_PV(profit, array_discounting[idx])
         
-        array_PV.append(cur_PV)
+        array_PV.append(round(cur_PV))
         
     return array_CF, array_PV
 
 
-def get_array_NPV():
-    array_NPV = [array_PV[0]]
-    for idx in range(1, Number_of_months + 1, 1):
-        array_NPV.append(array_NPV[-1] + array_PV[idx])
-    return array_NPV
-
+def get_id_first_positive_NPV(array_NPV):
+    for idx, npv in enumerate(array_NPV):
+        if npv > 0:
+            return idx
+    return len(array_NPV)
+    
     
 if st.button('Расчет эффективности'):
     Number_pallets, Number_machines, Number_employees = get_number_pallets_and_machines_employees()
@@ -207,13 +207,12 @@ if st.button('Расчет эффективности'):
     array_NPV = get_array_NPV()
     
     IRR = np_fin.irr(array_CF)
-
     
     st.write(f'Необходимое кол-во ричтраков {Number_machines}')
     st.write(f'Необходимое кол-во сотрудников {Number_employees}')
     
     st.write(f'NPV за {Number_of_months} месяцев:', array_NPV[-1])
-    st.write(f'IRR на {Number_of_months} месяцев: {round(100 * IRR), 2}%')
+    st.write(f'IRR на {Number_of_months} месяцев: {round(100 * IRR, 2)}%')
     
     if array_NPV[-1] < 0:
         st.write(f'{Number_of_months} месяцев не хватает для окупаемости проекта')
@@ -226,4 +225,5 @@ if st.button('Расчет эффективности'):
     chart_data['PV'] = array_PV
     chart_data['NPV'] = array_NPV
 
-    st.bar_chart(chart_data)
+    st.bar_chart(chart_data)
+