Update src/map_viz_pred.py

src/map_viz_pred.py

@@ -42,10 +42,21 @@ def calling_pred_map_viz(counts_df1, input_day = None, input_hour = None):
     counts_df1.loc[:,'date_time'] = pd.to_datetime(counts_df1.loc[:,'date'] + " "+ counts_df1.loc[:,'time'], format='%Y-%m-%d %H:%M:%S')
     counts_df1.loc[:,'hour'] = counts_df1.loc[:,'date_time'].apply(hour_rounder)
     counts_df1.loc[:,'day_name'] = counts_df1.loc[:,'date_time'].dt.day_name()
-    filtered_date = counts_df1.iloc[-1]['date']
-    filtered_time = counts_df1.iloc[-1]['time']
-    filtered_day = counts_df1.iloc[-1]['day_name']
-    filtered_hour = counts_df1.iloc[-1]['hour']
+
+    if (input_day != None) & (input_hour != None):
+        filtered_day = input_day
+        filtered_hour = input_hour
+    else:
+        filtered_date = counts_df1.iloc[-1]['date']
+        filtered_time = counts_df1.iloc[-1]['time']
+        filtered_day = counts_df1.iloc[-1]['day_name']
+        filtered_hour = counts_df1.iloc[-1]['hour']
+
+    
+    # filtered_date = counts_df1.iloc[-1]['date']
+    # filtered_time = counts_df1.iloc[-1]['time']
+    # filtered_day = counts_df1.iloc[-1]['day_name']
+    # filtered_hour = counts_df1.iloc[-1]['hour']
 
 
     day_hour_view_group = counts_df1.groupby(by=['view', 'day_name', 'hour'])['total'].mean().reset_index()