Fixed database calls for new seed frame format

app.py

@@ -55,16 +55,31 @@ st.markdown("""
 def init_DK_seed_frames(sharp_split, slate_var):  
 
     if slate_var == 'Main Slate':
-        collection = db['DK_NHL_seed_frame_Main Slate']
+        collection = db['DK_NHL_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"DK_NHL_seed_frame_Main Slate"] 
     elif slate_var == 'Secondary Slate':
-        collection = db['DK_NHL_seed_frame_Secondary Slate']
+        collection = db['DK_NHL_Secondary_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"DK_NHL_seed_frame_Secondary Slate"] 
     elif slate_var == 'Auxiliary Slate':
-        collection = db['DK_NHL_seed_frame_Auxiliary Slate']
+        collection = db['DK_NHL_Late_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"DK_NHL_seed_frame_Late Slate"] 
 
     cursor = collection.find().limit(sharp_split)
 
     raw_display = pd.DataFrame(list(cursor))
     raw_display = raw_display[['C1', 'C2', 'W1', 'W2', 'W3', 'D1', 'D2', 'G', 'FLEX', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    dict_columns = ['C1', 'C2', 'W1', 'W2', 'W3', 'D1', 'D2', 'G', 'FLEX']
+    for col in dict_columns:
+        raw_display[col] = raw_display[col].map(names_dict)
     DK_seed = raw_display.to_numpy()
 
     return DK_seed
@@ -73,16 +88,34 @@ def init_DK_seed_frames(sharp_split, slate_var):
 def init_FD_seed_frames(sharp_split, slate_var):  
     
     if slate_var == 'Main Slate':
-        collection = db['FD_NHL_seed_frame_Main Slate']
+        collection = db['FD_NHL_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"FD_NHL_seed_frame_Main Slate"] 
+        cursor = collection.find().limit(sharp_split)
     elif slate_var == 'Secondary Slate':
-        collection = db['FD_NHL_seed_frame_Secondary Slate']
+        collection = db['FD_NHL_Secondary_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"FD_NHL_seed_frame_Secondary Slate"] 
+        cursor = collection.find().limit(sharp_split)
     elif slate_var == 'Auxiliary Slate':
-        collection = db['FD_NHL_seed_frame_Auxiliary Slate']
+        collection = db['FD_NHL_Late_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"FD_NHL_seed_frame_Late Slate"] 
+        cursor = collection.find().limit(sharp_split)
 
     cursor = collection.find().limit(sharp_split)
 
     raw_display = pd.DataFrame(list(cursor))
     raw_display = raw_display[['C1', 'C2', 'W1', 'W2', 'D1', 'D2', 'FLEX1', 'FLEX2', 'G', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    dict_columns = ['C1', 'C2', 'W1', 'W2', 'D1', 'D2', 'FLEX1', 'FLEX2', 'G']
+    for col in dict_columns:
+        raw_display[col] = raw_display[col].map(names_dict)
     FD_seed = raw_display.to_numpy()
 
     return FD_seed