Update app.py

app.py

@@ -49,16 +49,14 @@ def init_baselines():
     collection = db["DK_MLB_seed_frame"] 
     cursor = collection.find()
     
-    DK_seed = np.array(list(cursor))
-    # raw_display = pd.DataFrame(list(cursor))
-    # DK_seed = raw_display[['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'salary', 'proj']]
+    raw_display = pd.DataFrame(list(cursor))
+    DK_seed = raw_display[['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'salary', 'proj']]
 
     collection = db["FD_MLB_seed_frame"] 
     cursor = collection.find()
     
-    FD_seed = np.array(list(cursor))
-    # raw_display = pd.DataFrame(list(cursor))
-    # FD_seed = raw_display[['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'salary', 'proj']]
+    raw_display = pd.DataFrame(list(cursor))
+    FD_seed = raw_display[['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'salary', 'proj']]
     
     worksheet = sh.worksheet('DK_Projections')
     load_display = pd.DataFrame(worksheet.get_all_records())
@@ -93,30 +91,30 @@ with tab1:
             
             team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
             if team_var1 == 'Specific Teams':
-                    team_var2 = st.multiselect('Which teams do you want?', options = np.unique(DK_seed[:, 2]))
+                    team_var2 = st.multiselect('Which teams do you want?', options = DK_seed['Team'].unique())
             elif team_var1 == 'Full Slate':
-                    team_var2 = np.unique(DK_seed[:, 2])
+                    team_var2 = DK_seed.Team.values.tolist()
             
             stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
             if stack_var1 == 'Specific Stack Sizes':
-                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = np.unique(DK_seed[:, 3]))
+                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = DK_seed['Team_count'].unique())
             elif stack_var1 == 'Full Slate':
-                    stack_var2 = np.unique(DK_seed[:, 3])
+                    stack_var2 = DK_seed.Team_count.values.tolist()
                     
         elif site_var1 == 'Fanduel':
             raw_baselines = fd_raw
             
             team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
             if team_var1 == 'Specific Teams':
-                    team_var2 = st.multiselect('Which teams do you want?', options = np.unique(FD_seed[:, 2]))
+                    team_var2 = st.multiselect('Which teams do you want?', options = FD_seed['Team'].unique())
             elif team_var1 == 'Full Slate':
-                    team_var2 = np.unique(FD_seed[:, 2])
+                    team_var2 = FD_seed.Team.values.tolist()
             
             stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
             if stack_var1 == 'Specific Stack Sizes':
-                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = np.unique(FD_seed[:, 3]))
+                    stack_var2 = st.multiselect('Which stack sizes do you want?', options = FD_seed['Team_count'].unique())
             elif stack_var1 == 'Full Slate':
-                    stack_var2 = np.unique(FD_seed[:, 3])
+                    stack_var2 = FD_seed.Team_count.values.tolist()