Refactor league data loading and simplify initialization process

app.py

@@ -40,16 +40,14 @@ def init_conn():
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
         db = client["League_of_Legends"]
      
-        LCK_Data = st.secrets['LOL_Data']
-        LEC_Data = st.secrets['LEC_Data']
-        LCS_Data = st.secrets['LCS_Data']
+        LOL_Data = st.secrets['LOL_Data']
 
         gc = gspread.service_account_from_dict(credentials)
         gc2 = gspread.service_account_from_dict(credentials2)
 
-        return gc, gc2, db, LCK_Data, LEC_Data, LCS_Data
+        return gc, gc2, db, LOL_Data
     
-gcservice_account, gcservice_account2, db, LCK_Data, LEC_Data, LCS_Data = init_conn()
+gcservice_account, gcservice_account2, db, LOL_Data = init_conn()
 
 percentages_format = {'Exposure': '{:.2%}'}
 freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
@@ -77,25 +75,12 @@ def init_DK_seed_frames():
         return DK_seed
 
 @st.cache_data(ttl = 599)
-def init_baselines(league_data):
-
-    if league_data == 'LCK':
-        try:
-            sh = gcservice_account.open_by_url(LCK_Data)
-        except:
-            sh = gcservice_account2.open_by_url(LCK_Data)
-    elif league_data == 'LEC':
-        try:
-            sh = gcservice_account.open_by_url(LEC_Data)
-        except:
-            sh = gcservice_account2.open_by_url(LEC_Data)
-    elif league_data == 'LCS':
-        try:
-            sh = gcservice_account.open_by_url(LCS_Data)
-        except:
-            sh = gcservice_account2.open_by_url(LCS_Data)
+def init_baselines():
+    try:
+        sh = gcservice_account.open_by_url(LOL_Data)
+    except:
+        sh = gcservice_account2.open_by_url(LOL_Data)
     
-
     worksheet = sh.worksheet('ROO')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
@@ -170,12 +155,12 @@ with tab2:
               for key in st.session_state.keys():
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames()
-              dk_raw = init_baselines('LCK')
+              dk_raw = init_baselines()
               
         slate_var1 = st.radio("Which data are you loading?", ('LCK', 'LEC', 'LCS'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings'))
         if site_var1 == 'Draftkings':
-            raw_baselines = init_baselines(slate_var1)
+            raw_baselines = dk_raw[dk_raw['league'] == slate_var1]
             column_names = dk_columns
             
             team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
@@ -224,17 +209,18 @@ with tab1:
               for key in st.session_state.keys():
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames()
-              dk_raw = init_baselines('LCK')
+              dk_raw = init_baselines()
         sim_slate_var1 = st.radio("Which data are you loading?", ('LCK', 'LEC', 'LCS'), key='sim_slate_var1')
 
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings'), key='sim_site_var1')
         if sim_site_var1 == 'Draftkings':
-            raw_baselines = init_baselines(sim_slate_var1)
+            raw_baselines = dk_raw[dk_raw['league'] == sim_slate_var1]
             column_names = dk_columns
             
         contest_var1 = st.selectbox("What contest size are you simulating?", ('Small', 'Medium', 'Large', 'Custom'))
         if contest_var1 == 'Small':
             Contest_Size = 1000
+
         elif contest_var1 == 'Medium':
             Contest_Size = 5000
         elif contest_var1 == 'Large':