Update main.py

main.py

@@ -17,7 +17,7 @@ app = FastAPI()
 #Root endpoints
 @app.get("/")
 def root():
-    return {"API": "Hail API 2020"}
+    return {"API": "ACPC HRRR"}
     
 def geocode_address(address):
 
@@ -50,16 +50,15 @@ def get_hail_data(address, start_date, end_date, radius_miles, get_max):
     lat, lon= geocode_address(address)
     
     # Convert Lat Lon to row & col on Array
-    transform = pickle.load(open('Data/transform_mrms.pkl', 'rb'))
-    
-    row, col = rasterio.transform.rowcol(transform, lon, lat)
-
-    files = [
-        'Data/2023_hail.h5',
-        'Data/2022_hail.h5',
-        'Data/2021_hail.h5',
-        'Data/2020_hail.h5'
-    ]
+    transform = pickle.load(open('Data/hrrr_crs.pkl', 'rb'))
+    row, col = rasterio.transform.rowcol(transform['affine'], lon, lat)
+
+
+    files = ["Data/APCP_2024_hrrr_v2.h5",
+        "Data/APCP_2020_hrrr_v3.h5",
+        "Data/APCP_2021_hrrr_3.h5",
+        "Data/APCP_2022_hrrr_v2.h5",
+        "Data/APCP_2023_hrrr_v2c.h5"]
 
     files_choosen = [i for i in files if any(i for j in years if str(j) in i)]