import imdlib as imd

def main():
    start_yr = 2020
    end_yr = 2020
    variable = 'tmax'
    file_dir = './imd_data'
    
    print("Downloading...")
    # This will download the 2020 tmax gridded file from IMD
    imd.get_data(variable, start_yr, end_yr, fn_format='yearwise', file_dir=file_dir)
    print("Download complete.")
    
    print("Opening...")
    data = imd.open_data(variable, start_yr, end_yr, 'yearwise', file_dir)
    ds = data.get_xarray()
    print("Xarray Dataset:", ds)

if __name__ == "__main__":
    main()