import csv
import numpy as np

data={}
with open('data/passenger.csv', 'r', encoding='utf-8') as f:
    reader = csv.reader(f)
    next(reader)  # skip header row
    for row in reader:
        data[int(row[0])] = np.sum(np.array(row[1:4], dtype=int))

dataList = []
for i in range(2015,2020):
    # print(data[i]/data[i-1])
    dataList.append(data[i]/data[i-1])
# print(data[2023]/data[2022])
dataList.append(data[2023]/data[2022])

dataList = np.array(dataList)
avgGrowth = np.mean(dataList)

prediction = data[2023]*((avgGrowth)**(2025-2023))
print("predicting 2025 total passengers when maintaining current taxation rate:",prediction)

def predictTotalPassengers(taxationRate):
    return (prediction/(0.946**(1.09/0.1)))*(0.946**(taxationRate/0.1))