Update main.py

main.py

@@ -46,18 +46,20 @@ 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
-    lat, lon= geocode_address(address)
-
+    try:
+        lat, lon= geocode_address(address)
+    except:
+        lat, lon= None,None
 
     
     # Convert Lat Lon to row & col on Array
@@ -126,7 +128,7 @@ def get_data(address, start_date, end_date, radius_miles, get_max):
     # df_data['Date'] = df_data['Date'].dt.strftime('%Y-%m-%d:%H')
 
     # return df_data
-    return lat, lon, crs_dic, row, col
+    return lat, lon, row, col
 
 @app.get('/APCP_Docker_Data')
 async def predict(address: str, start_date: str, end_date: str, radius_miles: int, get_max: bool):