Refactor app.py: Update seed frame initialization functions to accept a load size parameter, limiting the number of records fetched from the database. This change enhances performance by allowing for more controlled data retrieval in DraftKings and FanDuel seed frame functions.

app.py

@@ -22,10 +22,10 @@ dk_columns = ['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'FLEX', 'salary', 'proj', '
 fd_columns = ['PG1', 'PG2', 'SG1', 'SG2', 'SF1', 'SF2', 'PF1', 'PF2', 'C1', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 
 @st.cache_data(ttl = 60)
-def init_DK_seed_frames():  
+def init_DK_seed_frames(load_size):  
     
         collection = db["DK_NBA_seed_frame"] 
-        cursor = collection.find()
+        cursor = collection.find().limit(load_size)
     
         raw_display = pd.DataFrame(list(cursor))
         raw_display = raw_display[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'FLEX', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -34,10 +34,10 @@ def init_DK_seed_frames():
         return DK_seed
 
 @st.cache_data(ttl = 60)
-def init_DK_secondary_seed_frames():  
+def init_DK_secondary_seed_frames(load_size):  
     
         collection = db["DK_NBA_Secondary_seed_frame"] 
-        cursor = collection.find()
+        cursor = collection.find().limit(load_size)
     
         raw_display = pd.DataFrame(list(cursor))
         raw_display = raw_display[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'FLEX', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -46,10 +46,10 @@ def init_DK_secondary_seed_frames():
         return DK_secondary
 
 @st.cache_data(ttl = 60)
-def init_FD_seed_frames():  
+def init_FD_seed_frames(load_size):  
     
         collection = db["FD_NBA_seed_frame"] 
-        cursor = collection.find()
+        cursor = collection.find().limit(load_size)
     
         raw_display = pd.DataFrame(list(cursor))
         raw_display = raw_display[['PG1', 'PG2', 'SG1', 'SG2', 'SF1', 'SF2', 'PF1', 'PF2', 'C1', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -58,10 +58,10 @@ def init_FD_seed_frames():
         return FD_seed
 
 @st.cache_data(ttl = 60)
-def init_FD_secondary_seed_frames():  
+def init_FD_secondary_seed_frames(load_size):  
     
         collection = db["FD_NBA_Secondary_seed_frame"] 
-        cursor = collection.find()
+        cursor = collection.find().limit(load_size)
     
         raw_display = pd.DataFrame(list(cursor))
         raw_display = raw_display[['PG1', 'PG2', 'SG1', 'SG2', 'SF1', 'SF2', 'PF1', 'PF2', 'C1', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]