Inital Commit (Creates Graphs)

This commit is contained in:
Venrey 2024-07-05 22:46:20 +00:00
commit 435a2a453a
30 changed files with 348 additions and 0 deletions

102
main.py Normal file
View file

@ -0,0 +1,102 @@
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')