Update app.py

app.py

@@ -3,7 +3,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 import math
-import numpy_financial as np_fin 
+import numpy_financial as np_fin
 
 st.title("Калькулятор Toyota📄")
 
@@ -31,7 +31,7 @@ if genre == 'Покупка автоматизированного ричтра
     Income_tax_rate = st.number_input('Ставка налога на прибыль', value=0.2)
     Number_of_autopilots_serviced_by_one_employee = st.number_input('Количество автопилотов, обслуживаемых 1 работником', value=10)
     discounting = st.number_input('Ставка дисконтирования в месяц', value=0.028) 
-        
+    
 elif genre == 'Аренда автоматизированного ричтрака':
     Rent_an_autopilot = st.number_input('Аренда автопилота, руб.', value = 20000)
     Rent_of_peripheral_equipment = st.number_input('Аренда перефирийного оборудования, руб.', value = 4000)
@@ -68,8 +68,8 @@ else:
     Number_of_working_days_month = st.number_input('Количество рабочих дней в месяц, дн.', value=20)
     Social_benefits_for_one_employee = st.number_input('Социальные выплаты 1 сотруднику, руб.', value=10000)
     Insurance_rate = st.number_input('Ставка страхования', value=0.005)
-    The_cost_of_demaged_goods = st.number_input('Стоимость поврежденного товара', value=1000)
-    Product_damage_rate= st.number_input('Коэфициент повреждения товара', value=0.005)
+    The_cost_of_demaged_goods = ('Стоимость поврежденного товара', value=1000)
+    Product_damage_rate= ('Коэфициент повреждения товара', value=0.005)
     discounting = st.number_input('Ставка дисконтирования в месяц', value=0.028)
 
 
@@ -100,31 +100,41 @@ def get_revenue(idx):
 
 def get_costs(idx):
     indexation = math.floor(idx / 12)
-    
+    start_year = ((idx - 1) % 12 == 0)
+    cost = 0
     wage_fund_with_indexation = (Social_benefits_for_one_employee + Monthly_salary_rate) * math.pow(Rate_CPI, indexation)
     
     if genre == 'Покупка автоматизированного ричтрака':
         Expected_repair_costs_per_month = Number_machines * Equipment_breakdown_rate * The_cost_of_repairing_one_car
         Wage_Fund = Number_employees * wage_fund_with_indexation
-        The_cost_of_insurance = Number_machines * Insurance_rate * (Buying_an_autopilot + Purchase_of_peripheral_equipment)
         
-        cost = Purchase_by + Number_machines * Maintenance + Expected_repair_costs_per_month + Wage_Fund + The_cost_of_insurance
+        if start_year:
+            The_cost_of_insurance = Number_machines * Insurance_rate * (Buying_an_autopilot + Purchase_of_peripheral_equipment)
+            cost += The_cost_of_insurance + Purchase_by
+            
+        cost += Number_machines * Maintenance + Expected_repair_costs_per_month + Wage_Fund
         
     elif genre == 'Аренда автоматизированного ричтрака':
         Expected_repair_costs_per_month = Number_machines * Equipment_breakdown_rate * The_cost_of_repairing_one_car
         Wage_Fund = Number_employees * wage_fund_with_indexation
         Autopilot_rental = Number_machines * Rent_an_autopilot
         
-        cost = Autopilot_rental + Rent_of_peripheral_equipment + Purchase_by + Number_machines * Maintenance + Wage_Fund + Expected_repair_costs_per_month
+        if start_year:
+            cost += Purchase_by
+        
+        cost += Autopilot_rental + Rent_of_peripheral_equipment + Purchase_by + Number_machines * Maintenance + Wage_Fund + Expected_repair_costs_per_month
         
     else:
         Expected_repair_costs_per_month = Number_machines * Equipment_breakdown_rate * The_cost_of_repairing_one_car
         Wage_Fund = Number_employees * wage_fund_with_indexation
-        The_cost_of_insurance = Number_machines * Insurance_rate * Buying_a_richtruck
         Expected_costs_for_the_purchase_of_damaged_goods = Efficiency * The_cost_of_demaged_goods * Product_damage_rate
         
-        cost = Number_machines * Maintenance + Expected_repair_costs_per_month + Wage_Fund + The_cost_of_insurance + Expected_costs_for_the_purchase_of_damaged_goods
-    
+        if start_year:
+            The_cost_of_insurance = Number_machines * Insurance_rate * Buying_a_richtruck
+            cost += The_cost_of_insurance
+            
+        cost += Number_machines * Maintenance + Expected_repair_costs_per_month + Wage_Fund + Expected_costs_for_the_purchase_of_damaged_goods
+        
     return cost
 
 def get_profit(amortization, idx):
@@ -189,15 +199,13 @@ 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('Расчет эффективности'):
     Number_pallets, Number_machines, Number_employees = get_number_pallets_and_machines_employees()
     array_CF, array_PV = get_array_CF_PV()
@@ -222,5 +230,4 @@ if st.button('Расчет эффективности'):
     chart_data['PV'] = array_PV
     chart_data['NPV'] = array_NPV
 
-    st.line_chart(chart_data)
-        
+    st.line_chart(chart_data)