Update app.py

app.py

@@ -224,14 +224,13 @@ def get_array_CF_PV():
         
         array_PV.append(round(cur_PV))
         
+    sale_price_mashine = 0
     if genre == 'Покупка автоматизированного ричтрака':
-        array_PV.append(0.7*(I_0 - np.sum(array_amortization)) / array_discounting[-1])
-        array_CF.append(0.7*(I_0 - np.sum(array_amortization)))
+        sale_price_mashine = 0.7*(I_0 - np.sum(array_amortization)) / array_discounting[-1]
     elif genre == 'Покупка механизированного ричтрака':
-        array_PV.append(0.7*(I_0 - np.sum(array_amortization)) / array_discounting[-1])
-        array_CF.append(0.7*(I_0 - np.sum(array_amortization)))
+        sale_price_mashine = 0.7*(I_0 - np.sum(array_amortization)) / array_discounting[-1]
         
-    return array_CF, array_PV
+    return array_CF, array_PV, sale_price_mashine
 
 
 def get_array_NPV():
@@ -250,7 +249,7 @@ def get_id_first_positive_NPV(array_NPV):
     
 if st.button('Расчет эффективности'):
     Number_pallets, Number_machines, Number_employees = get_number_pallets_and_machines_employees()
-    array_CF, array_PV = get_array_CF_PV()
+    array_CF, array_PV, sale_price_mashine = get_array_CF_PV()
     
     array_NPV = get_array_NPV()
     
@@ -259,6 +258,11 @@ if st.button('Расчет эффективности'):
     st.write(f'Необходимое кол-во ричтраков {Number_machines}')
     if genre == "Покупка механизированного ричтрака":
         st.write(f'Необходимое кол-во сотрудников {Number_employees}')
+        
+    if genre == 'Покупка автоматизированного ричтрака':
+        st.write(f'С учетом изнашивания машины можно продать за {rount(sale_price_mashine)}, в конце проекта')
+    elif genre == 'Покупка механизированного ричтрака':
+        st.write(f'С учетом изнашивания машины можно продать за {rount(sale_price_mashine)}, в конце проекта')
     
     st.write(f'NPV за {Number_of_months} месяцев:', array_NPV[-1])
     st.write(f'IRR на {Number_of_months} месяцев: {round(100 * IRR, 2)}%')