import pandas as pd, json
df = pd.read_excel('/Users/maxim.ziatdinov/Code/SciLink/planning_session_20260319_180040/knowledge/PWSdatabase.xlsx')
element_cols = ['Ag', 'Al', 'As', 'Au', 'B', 'BO3', 'Ba', 'Be', 'Bi', 'Br', 'CO3', 'HCO3', 'Ca', 'Cd', 'Cl', 'Co', 'Cr', 'Cs', 'Cu', 'F', 'FeTot', 'FeIII', 'FeII', 'FeS', 'FeAl', 'FeAl2O3', 'Hg', 'I', 'K', 'KNa', 'Li', 'Mg', 'Mn', 'Mo', 'N', 'NO2', 'NO3', 'NO3NO2', 'NH4', 'TKN', 'Na', 'Ni', 'OH', 'P', 'PO4', 'Pb', 'Rh', 'Rb', 'S', 'SO3', 'SO4', 'HS', 'Sb', 'Sc', 'Se', 'Si', 'Sn', 'Sr', 'Ti', 'Tl', 'U', 'V', 'W', 'Zn']
answer = element_cols
summary = f"There are {len(element_cols)} columns containing element/ion concentration data, including major ions (Na, Ca, Mg, K, Cl, SO4, HCO3), trace elements (Li, Sr, Ba, B, Mn, Fe, etc.), and nutrients (NO3, NH4, PO4, etc.)."
print(json.dumps({"answer": answer, "summary": summary}))
