Update app.py

app.py

@@ -1,10 +1,3 @@
-import streamlit as st
-import matplotlib.pyplot as plt
-import numpy as np
-import pandas as pd
-import math
-import numpy_financial as npf
-
 import streamlit as st
 import matplotlib.pyplot as plt
 import numpy as np
@@ -16,27 +9,27 @@ st.title("Калькулятор Toyota📄")
 genre = st.radio("Какой из вариантов", ("Покупка автоматизированного ричтрака", "Аренда автоматизированного ричтрака", "Покупка механизированного ричтрака"))
 
 if genre == 'Покупка автоматизированного ричтрака':
-    Buying_an_autopilot = st.number_input('Покупка автопилота, руб.', value=2000000)
-    Purchase_of_peripheral_equipment = st.number_input('Покупка перефирийного оборудования, руб.', value=40000)
-    Purchase_by = st.number_input('Покупка ПО, руб.', value=10000)
-    Introduction_of_autopilot = st.number_input('Внедрение автопилота, руб.', value=20000)
-    Maintenance = st.number_input('Обслуживание (месяц), руб.', value=100000)
-    The_cost_of_repairing_one_car = st.number_input('Стоимость ремонта одной машины, руб.', value=10000)
-    Inflation_rate = st.number_input('Уровень инфляции', value=0.04)
-    Monthly_salary_rate = st.number_input('Заработная ставка, руб. в месяц', value=40000)
-    Number_of_months = st.number_input('Количество месяцев', value=12)
-    Equipment_breakdown_rate = st.number_input('Коэффициент поломки оборудования', value=0.1)
-    Price_for_processing_ont_pallet = st.number_input('Цена за обработку 1 паллета, руб.', value=300)
-    Cargo_flow = st.number_input('Грузопоток, шт/месяц', value=10000)
-    Efficiency = st.number_input('Производительность, шт в час', value=5)
-    Number_of_working_hours = st.number_input('Количество рабочих часов', value=8)
-    Shift_of_one_employee = st.number_input('Смена 1 работника, ч.', value=8)
-    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)
-    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)
+    Buying_an_autopilot = st.number_input('Покупка автопилота, руб.', value=0)
+    Purchase_of_peripheral_equipment = st.number_input('Покупка перефирийного оборудования, руб.')
+    Purchase_by = st.number_input('Покупка ПО, руб.')
+    Introduction_of_autopilot = st.number_input('Внедрение автопилота, руб.')
+    Maintenance = st.number_input('Обслуживание (месяц), руб.')
+    The_cost_of_repairing_one_car = st.number_input('Стоимость ремонта одной машины, руб.')
+    Monthly_salary_rate = st.number_input('Заработная ставка, руб. в месяц')
+    Number_of_months = st.number_input('Количество месяцев')
+    Equipment_breakdown_rate = st.number_input('Коэфициент поломки оборудования ')
+    Price_for_processing_ont_pallet = st.number_input('Цена за обработку 1 паллета, руб.')
+    Cargo_flow = st.number_input('Грузопоток, шт/месяц ')
+    Efficiency = st.number_input('Производительность, шт в час')
+    Number_of_working_hours = st.number_input('Количество рабочих часов')
+    Shift_of_one_employee = st.number_input('Смена 1 работника, ч.')
+    Number_of_working_days_month = st.number_input('Количество рабочих дней в месяц, дн.')
+    Social_benefits_for_one_employee = st.number_input('Социальные выплаты 1 сотруднику, руб.')
+    Insurance_rate = st.number_input('Ставка страхования')
+    Income_tax_rate = st.number_input('Ставка налога на прибыль')
+    Number_of_autopilots_serviced_by_one_employee = st.number_input('Количество автопилотов, обслуживаемых 1 работником')
+    discounting = st.number_input('Ставка дисконтирования в месяц')
+    
 elif genre == 'Аренда автома��изированного ричтрака':
     Rent_an_autopilot = st.number_input('Аренда автопилота, руб.', value = 20000)
     Rent_of_peripheral_equipment = st.number_input('Аренда перефирийного оборудования, руб.', value = 4000)
@@ -57,6 +50,7 @@ elif genre == 'Аренда автоматизированного ричтра
     Income_tax_rate = st.number_input('Ставка налога на прибыль', value = 0.2)
     Number_of_autopilots_serviced_by_one_employee = st.number_input('Количество автопилотов, обслуживаемых одним работником, шт.', value = 10)
     discounting = st.number_input('Ставка дисконтирования в месяц', value = 0.028)
+    
 else:
     Buying_a_richtruck = st.number_input('Покупка ричтрака, руб.', value=2000000)
     Maintenance = st.number_input('Обслуживание (месяц), руб.', value=10000)
@@ -72,20 +66,25 @@ 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)
 
 
-
-Rate_CPI = 1.1
+Rate_CPI = 1.05
 
 
 def get_number_pallets_and_machines_employees():
     Number_pallets = Efficiency * Number_of_working_hours * Number_of_working_days_month
     Number_machines = math.ceil(Cargo_flow / Number_pallets)
-    Number_employees = math.ceil((Number_machines * Number_of_working_hours / Shift_of_one_employee) / Number_of_autopilots_serviced_by_one_employee)
     
+    if genre == 'Покупка автоматизированного ричтрака':
+        Number_employees = math.ceil((Number_machines * Number_of_working_hours / Shift_of_one_employee) / Number_of_autopilots_serviced_by_one_employee)
+    elif genre == 'Аренда автоматизированного ричтрака':
+        Number_employees = math.ceil((Number_machines * Number_of_working_hours / Shift_of_one_employee) / Number_of_autopilots_serviced_by_one_employee)
+    else:
+        Number_employees = math.ceil((Number_machines * Number_of_working_hours / Shift_of_one_employee))
+   
     return Number_pallets, Number_machines, Number_employees
 
 
@@ -99,14 +98,32 @@ def get_revenue(idx):
 
 def get_costs(idx):
     indexation = math.floor(idx / 12)
+    
     wage_fund_with_indexation = math.pow((Social_benefits_for_one_employee + Monthly_salary_rate), indexation)
     
-    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)
+    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
+        
+    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
+        
+    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
     
-    return Purchase_by + Number_machines * Maintenance + Expected_repair_costs_per_month + Wage_Fund + The_cost_of_insurance
-
+    return cost
 
 def get_profit(amortization, idx):
     profit = get_revenue(idx) - get_costs(idx) - amortization
@@ -136,10 +153,18 @@ def get_amortization(value):
 
 
 def get_array_CF_PV():
-    I_0 = Buying_an_autopilot * Number_machines + Purchase_of_peripheral_equipment + Introduction_of_autopilot
-    
+    if genre == 'Покупка автоматизированного ричтрака':
+        I_0 = Buying_an_autopilot * Number_machines + Purchase_of_peripheral_equipment + Introduction_of_autopilot * Number_machines
+        amortizat = Buying_an_autopilot * Number_machines + Introduction_of_autopilot * Number_machines
+    elif genre == 'Аренда автоматизированного ричтрака':
+        I_0 = Rent_of_peripheral_equipment
+        amortizat = 0
+    else:
+        I_0 = Buying_a_richtruck * Number_machines
+        amortizat = Buying_a_richtruck * Number_machines
+        
     array_discounting = get_array_discounting()
-    array_amortization = get_amortization(I_0)
+    rray_amortization = get_amortization(amortizat)
     
     array_PV = [-I_0]
     array_CF = [-I_0]
@@ -179,3 +204,4 @@ if st.button('Сюда кликнуть чтобы посчитать'):
     chart_data['NPV'] = array_NPV
 
     st.line_chart(chart_data)
+