Update main.py

main.py

@@ -46,14 +46,14 @@ def geocode_address(address):
 
 def get_data(address, start_date, end_date, radius_miles, get_max):
 
-    # start_date = pd.Timestamp(str(start_date)).strftime('%Y%m%d%H')
-    # end_date = pd.Timestamp(str(end_date)).strftime('%Y%m%d%H')
-    # date_years = pd.date_range(start=start_date[:-2], end=end_date[:-2], freq='M')
-    # date_range_days = pd.date_range(start_date[:-2], end_date[:-2], freq='H')
-    # years = list(set([d.year for d in date_years]))
-
-    # if len(years) == 0:
-    #     years = [pd.Timestamp(start_date).year]
+    start_date = pd.Timestamp(str(start_date)).strftime('%Y%m%d%H')
+    end_date = pd.Timestamp(str(end_date)).strftime('%Y%m%d%H')
+    date_years = pd.date_range(start=start_date[:-2], end=end_date[:-2], freq='M')
+    date_range_days = pd.date_range(start_date[:-2], end_date[:-2], freq='H')
+    years = list(set([d.year for d in date_years]))
+
+    if len(years) == 0:
+        years = [pd.Timestamp(start_date).year]
     
     # Geocode Address
     try:
@@ -69,15 +69,15 @@ def get_data(address, start_date, end_date, radius_miles, get_max):
         row=col=None
    
 
-    # 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 = [
+        # '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)]
+    files_choosen = [i for i in files if any(i for j in years if str(j) in i)]
 
 
     # # Query and Collect H5 Data