Update main.py

main.py

@@ -18,6 +18,17 @@ app = FastAPI()
 @app.get("/")
 def root():
     return {"API": "ACPC HRRR"}
+
+
+def lat_lon_to_row_col(lat, lon):
+    crs_dic = pickle.load(open('Data/hrrr_crs.pkl', 'rb'))
+    transform = crs_dic['affine']
+    trans_rtma = crs_dic['proj_4326']
+    lon_rtma, lat_rtma = trans_rtma.transform(lon, lat)
+
+    row, col = rasterio.transform.rowcol(transform, lon_rtma, lat_rtma)
+    row, col = int(row), int(col)
+    return row, col
     
 def geocode_address(address):
 
@@ -54,19 +65,10 @@ def get_data(address, start_date, end_date, radius_miles, get_max):
     
     # Convert Lat Lon to row & col on Array
     try:
-        crs_dic = pickle.load(open('Data/hrrr_crs.pkl', 'rb'))
-    except:
-        crs_dic=None
-    try:
-        transform = crs_dic['affine']
-        trans_hrrr = crs_dic['proj_4326']
-        lon_hrrr, lat_hrrr = trans_hrrr.transform(lon, lat)
-
-        row, col = rasterio.transform.rowcol(transform, lon_hrrr, lat_hrrr)
-        row, col = int(row), int(col)
+        row, col = lat_lon_to_row_col(lat, lon)
     except:
         row=col=None
-
+   
 
     # files = [
     #     # 'Data/APCP_2024_hrrr_v2.h5',