Update main.py

main.py

@@ -37,8 +37,8 @@ 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))
+    # 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')
@@ -81,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:row +
-                             radius+1, col-radius:col+radius+1]
+            data = f['hail'][date_idx, row-radius_miles:row +
+                             radius_miles+1, col-radius_miles:col+radius_miles+1]
 
             all_data.append(data)
             all_dates.append(dates)
@@ -94,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) == 1, True, False)
+    disk_mask = np.where(disk(radius_miles) == 1, True, False)
     data_mat = np.where(disk_mask, data_mat, -1).round(3)
 
     # Process to DataFrame