Update main.py

main.py

@@ -37,6 +37,10 @@ def geocode_address(address):
 
 def get_hail_data(address, start_date, end_date, radius_miles, get_max):
 
+    resolution=1 # mrms 1 and hrrr is 3
+    radius = int(np.ceil(radius_miles*1.6/resolution))
+
+    
     start_date = pd.Timestamp(str(start_date)).strftime('%Y%m%d')
     end_date = pd.Timestamp(str(end_date)).strftime('%Y%m%d')
     date_years = pd.date_range(start=start_date, end=end_date, freq='M')
@@ -77,8 +81,8 @@ def get_hail_data(address, start_date, end_date, radius_miles, get_max):
 
             # Select Data by Date and Radius
             dates = dates[date_idx]
-            data = f['hail'][date_idx, row-radius_miles:row +
-                             radius_miles+1, col-radius_miles:col+radius_miles+1]
+            data = f['hail'][date_idx, row-radius:row +
+                             radius+1, col-radius:col+radius+1]
 
             all_data.append(data)
             all_dates.append(dates)
@@ -90,7 +94,7 @@ def get_hail_data(address, start_date, end_date, radius_miles, get_max):
     data_mat = np.where(data_all < 0, 0, data_all)*0.0393701
 
     # Get Radius of Data
-    disk_mask = np.where(disk(radius_miles) == 1, True, False)
+    disk_mask = np.where(disk(radius) == 1, True, False)
     data_mat = np.where(disk_mask, data_mat, -1).round(3)
 
     # Process to DataFrame