Update app.py

app.py

@@ -189,18 +189,35 @@ def get_array_NPV():
     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('Сюда кликнуть чтобы посчитать'):
+if st.button('Расчет эффективности'):
     Number_pallets, Number_machines, Number_employees = get_number_pallets_and_machines_employees()
     array_CF, array_PV = get_array_CF_PV()
     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} месяцев:', IRR)
     
+    if array_NPV[-1] < 0:
+        st.write(f'{Number_of_months} месяцев не хватает для окупаемости проекта')
+    else:
+        id_first_positive_NPV = get_id_first_positive_NPV(array_NPV)
+        st.write(f'Проект окупается на {id_first_positive_NPV} месяц')
+        
+    
     chart_data = pd.DataFrame(columns=['PV', 'NPV'])
     chart_data['PV'] = array_PV
     chart_data['NPV'] = array_NPV