from openpyxl import Workbook, load_workbook
import numpy as np
import os
import shutil
import graphs
import cords
from PIL import Image


#Setup directorys
try:
    os.mkdir('./.gen')
    os.mkdir('./gen')
except:
    pass

# Generate Graphs
wb = load_workbook('./Numbers.xlsx', data_only=True)
graphs.gen_EF(wb)
graphs.gen_LI(wb)
graphs.gen_DI(wb)
graphs.gen_Roth_IRA(wb)
graphs.gen_Roth_Catch_Up(wb)
graphs.gen_401K_Contribution(wb)
graphs.gen_401K_Catch_Up(wb)
graphs.gen_HSA_Contribution(wb)
graphs.gen_FSA_Contribution(wb)
graphs.gen_Oh_I_Mountain(wb)


### Image Work

## I Mountain image code
I_Mountain = Image.open('./templets/I_mountain.png').convert('RGBA').resize((1920, 1080))
I_Mountain_Mask = Image.open('./templets/mask/I_mountain_mask.png').convert('L')
blank = Image.open('./templets/I_mountain.png').convert('RGBA').resize((1920, 1080))

EF = Image.open('./.gen/EF.png')
EF = cords.scale(EF,0.265)
blank.paste(EF, cords.center_on(EF,(463,805)),EF)

DI = Image.open('./.gen/DI.png')
DI = cords.scale(DI,0.265)
blank.paste(DI, cords.center_on(DI,(698,450)),DI)

LI = Image.open('./.gen/LI.png')
LI = cords.scale(LI,0.265)
blank.paste(LI, cords.center_on(LI,(1472,802)),LI)

#Mask and combine
Image.composite(blank, I_Mountain, I_Mountain_Mask).save("./gen/I_mountain.png")


## O Mountain Image Code
O_Mountain = Image.open('./templets/O_mountain.png').convert('RGBA').resize((1920, 1080))
O_Mountain_Mask = Image.open('./templets/mask/O_mountain_mask.png').convert('L')
O_Mountain_Mask1 = Image.open('./templets/mask/O_mountain_mask1.png').convert('L')
O_Mountain_Mask2 = Image.open('./templets/mask/O_mountain_mask2.png').convert('L')
blank = Image.open('./templets/O_mountain.png').convert('RGBA').resize((1920, 1080))
blank2 = Image.new('RGBA',(1920,1080))

K401 = Image.open('./.gen/401K_Contribution.png')
K401 = cords.scale(K401,0.378)
blank.paste(K401, cords.center_on(K401,(386,806)),K401)

K401C = Image.open('./.gen/401K_Catch_Up.png')
K401C = cords.scale(K401C,0.378)
blank2.paste(K401C, cords.center_on(K401C,(557,806)),K401C)

IRAC = Image.open('./.gen/Roth_Catch_Up.png')
IRAC = cords.scale(IRAC,0.265)
blank.paste(IRAC, cords.center_on(IRAC,(1177,855)),IRAC)

IRA = Image.open('./.gen/Roth_IRA.png')
IRA = cords.scale(IRA,0.265)
blank2.paste(IRA, cords.center_on(IRA,(1425,855)),IRA)

# FSA = Image.open('./.gen/FSA_Contribution.png')
# FSA = cords.scale(FSA,0.265)
# blank2.paste(FSA, cords.center_on(FSA,(1425,855)),FSA)

blank2 = Image.composite(blank2, Image.new('RGBA',(1920,1080)), O_Mountain_Mask2)
blank = Image.alpha_composite(blank,blank2)

#blank.save('blank.png')
Image.composite(blank, O_Mountain, O_Mountain_Mask).save("./gen/O_mountian.png")


## Oh I Mountain Image Code
Oh_I_Mountain = Image.open('./templets/Oh_I_mountain.png').convert('RGBA').resize((1920, 1080))
Oh_I_Mountain_Mask = Image.open('./templets/mask/Oh_I_mountain_mask.png').convert('L')
blank = Image.open('./templets/Oh_I_mountain.png').convert('RGBA').resize((1920, 1080))

Goal = Image.open('./.gen/Monthly_Income_Goal.png')
Goal = cords.scale(Goal,0.68)
blank.paste(Goal, cords.center_on(Goal,(946,673)),Goal)

#Mask and combine
Image.composite(blank, Oh_I_Mountain, Oh_I_Mountain_Mask).save("./gen/Oh_I_mountain.png")

#Remove directorys
shutil.rmtree('./.gen')