Update src/map_viz_pred.py

src/map_viz_pred.py

@@ -11,8 +11,11 @@ import requests
 import pandas as pd
 import datetime
 
-def calling_pred_map_viz():
-    r = "snazzy-image-01.svg"
+import warnings
+warnings.filterwarnings("ignore")
+
+def calling_pred_map_viz(counts_df1, input_date = None, input_hour = None):
+    r = "svg/snazzy-image-01.svg"
     tree = etree.parse(r)
     root = tree.getroot()
     path_elems = root.findall('.//{http://www.w3.org/2000/svg}path')
@@ -35,7 +38,6 @@ def calling_pred_map_viz():
             continue
 
 
-    counts_df1 = merge_volumes(counts_df)
     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()
@@ -86,14 +88,9 @@ def calling_pred_map_viz():
 
     fig = plt.figure(figsize=(10,10))
     ax = fig.add_subplot(111)
-    # ax.close()
     collection.set_transform(ax.transData)
     ax.add_artist(collection)
     ax.set_xlim([100, 1900])
     ax.set_ylim([1800, 0])
-    ax.text(200, 100, '1,000,000', color='r', size='large', name = 'Georgia')
-    ax.set_title(datetime.datetime.strptime(last_date, '%Y-%m-%d').strftime('%a, %d %B %Y') + " | " + last_time + " SGT", fontname = 'Georgia')
-    plt.show()
-
-
-calling_map_viz()
+    ax.set_title(datetime.datetime.strptime(filtered_day + " | " + filtered_hour + " SGT", fontname = 'Georgia')
+    return fig