Update app.py

app.py

@@ -12,6 +12,46 @@ import gspread
 import random
 import gc
 
+def init_conn():
+          scope = ['https://www.googleapis.com/auth/spreadsheets',
+                    "https://www.googleapis.com/auth/drive"]
+          
+          credentials = {
+            "type": "service_account",
+            "project_id": "sheets-api-connect-378620",
+            "private_key_id": "1005124050c80d085e2c5b344345715978dd9cc9",
+            "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCtKa01beXwc88R\nnPZVQTNPVQuBnbwoOfc66gW3547ja/UEyIGAF112dt/VqHprRafkKGmlg55jqJNt\na4zceLKV+wTm7vBu7lDISTJfGzCf2TrxQYNqwMKE2LOjI69dBM8u4Dcb4k0wcp9v\ntW1ZzLVVuwTvmrg7JBHjiSaB+x5wxm/r3FOiJDXdlAgFlytzqgcyeZMJVKKBQHyJ\njEGg/1720A0numuOCt71w/2G0bDmijuj1e6tH32MwRWcvRNZ19K9ssyDz2S9p68s\nYDhIxX69OWxwScTIHLY6J2t8txf/XMivL/636fPlDADvBEVTdlT606n8CcKUVQeq\npUVdG+lfAgMBAAECggEAP38SUA7B69eTfRpo658ycOs3Amr0JW4H/bb1rNeAul0K\nZhwd/HnU4E07y81xQmey5kN5ZeNrD5EvqkZvSyMJHV0EEahZStwhjCfnDB/cxyix\nZ+kFhv4y9eK+kFpUAhBy5nX6T0O+2T6WvzAwbmbVsZ+X8kJyPuF9m8ldcPlD0sce\ntj8NwVq1ys52eosqs7zi2vjt+eMcaY393l4ls+vNq8Yf27cfyFw45W45CH/97/Nu\n5AmuzlCOAfFF+z4OC5g4rei4E/Qgpxa7/uom+BVfv9G0DIGW/tU6Sne0+37uoGKt\nW6DzhgtebUtoYkG7ZJ05BTXGp2lwgVcNRoPwnKJDxQKBgQDT5wYPUBDW+FHbvZSp\nd1m1UQuXyerqOTA9smFaM8sr/UraeH85DJPEIEk8qsntMBVMhvD3Pw8uIUeFNMYj\naLmZFObsL+WctepXrVo5NB6RtLB/jZYxiKMatMLUJIYtcKIp+2z/YtKiWcLnwotB\nWdCjVnPTxpkurmF2fWP/eewZ+wKBgQDRMtJg7etjvKyjYNQ5fARnCc+XsI3gkBe1\nX9oeXfhyfZFeBXWnZzN1ITgFHplDznmBdxAyYGiQdbbkdKQSghviUQ0igBvoDMYy\n1rWcy+a17Mj98uyNEfmb3X2cC6WpvOZaGHwg9+GY67BThwI3FqHIbyk6Ko09WlTX\nQpRQjMzU7QKBgAfi1iflu+q0LR+3a3vvFCiaToskmZiD7latd9AKk2ocsBd3Woy9\n+hXXecJHPOKV4oUJlJgvAZqe5HGBqEoTEK0wyPNLSQlO/9ypd+0fEnArwFHO7CMF\nycQprAKHJXM1eOOFFuZeQCaInqdPZy1UcV5Szla4UmUZWkk1m24blHzXAoGBAMcA\nyH4qdbxX9AYrC1dvsSRvgcnzytMvX05LU0uF6tzGtG0zVlub4ahvpEHCfNuy44UT\nxRWW/oFFaWjjyFxO5sWggpUqNuHEnRopg3QXx22SRRTGbN45li/+QAocTkgsiRh1\nqEcYZsO4mPCsQqAy6E2p6RcK+Xa+omxvSnVhq0x1AoGAKr8GdkCl4CF6rieLMAQ7\nLNBuuoYGaHoh8l5E2uOQpzwxVy/nMBcAv+2+KqHEzHryUv1owOi6pMLv7A9mTFoS\n18B0QRLuz5fSOsVnmldfC9fpUc6H8cH1SINZpzajqQA74bPwELJjnzrCnH79TnHG\nJuElxA33rFEjbgbzdyrE768=\n-----END PRIVATE KEY-----\n",
+            "client_email": "gspread-connection@sheets-api-connect-378620.iam.gserviceaccount.com",
+            "client_id": "106625872877651920064",
+            "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+            "token_uri": "https://oauth2.googleapis.com/token",
+            "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+            "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/gspread-connection%40sheets-api-connect-378620.iam.gserviceaccount.com"
+          }
+
+          gc = gspread.service_account_from_dict(credentials)
+          return gc
+
+gspreadcon = init_conn()
+
+dk_player_url = 'https://docs.google.com/spreadsheets/d/1Yq0vGriWK-bS79e-bD6_u9pqrYE6Yrlbb_wEkmH-ot0/edit#gid=172632260'
+solver_conn = 'https://docs.google.com/spreadsheets/d/1H7kdaxVF7Bv3kb1DSa_3Dq6OaC9ajq9UAQfVyDluXzk/edit#gid=0'
+
+@st.cache_resource(ttl = 600)
+def init_baslines():
+    sh = gspreadcon.open_by_url(dk_player_url)
+    worksheet = sh.worksheet('DK_Salaries')
+    raw_display = pd.DataFrame(worksheet.get_all_records())
+    raw_display.rename(columns={"name": "Player"}, inplace = True)
+    raw_display['player_id_name'] = raw_display['player_id_name'] + " (" + raw_display['player_id'] + ")"
+    dk_ids = dict(zip(raw_display.Player, raw_display.player_id_name))
+    
+    return dk_ids
+
+dk_ids = init_baslines()
+
+freq_format = {'Proj Own': '{:.2%}', 'Freq': '{:.2%}'}
+
 tab1, tab2 = st.tabs(['Uploads', 'Manage Portfolio'])
 
 with tab1:
@@ -136,11 +176,13 @@ with tab1:
                                  
                             display_portfolio = split_portfolio[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'UTIL', 'Salary', 'Projection', 'Ownership']]
                             st.session_state.display_portfolio = display_portfolio
+                            st.session_state.export_portfolio = st.session_state.display_portfolio.replace(dk_ids, inplace=True)
                             hold_portfolio = display_portfolio.sort_values(by='Projection', ascending=False)
                             
                             st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:8].values, return_counts=True)),
                                                         columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
                             st.session_state.player_freq['Freq'] = st.session_state.player_freq['Freq'] / len(st.session_state.display_portfolio)
+                            st.session_state.player_freq['Proj Own'] = st.session_state.player_freq['Player'].map(player_own_dict)
                             st.session_state.player_freq = st.session_state.player_freq.set_index('Player')
                             
                             gc.collect() 
@@ -158,6 +200,7 @@ with tab2:
                 st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:8].values, return_counts=True)),
                                             columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
                 st.session_state.player_freq['Freq'] = st.session_state.player_freq['Freq'] / len(st.session_state.display_portfolio)
+                st.session_state.player_freq['Proj Own'] = st.session_state.player_freq['Player'].map(player_own_dict)
                 st.session_state.player_freq = st.session_state.player_freq.set_index('Player')
         with col2:
             if st.button("Trim Lineups", key='trim1'):
@@ -173,6 +216,7 @@ with tab2:
                 st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:8].values, return_counts=True)),
                                             columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
                 st.session_state.player_freq['Freq'] = st.session_state.player_freq['Freq'] / len(st.session_state.display_portfolio)
+                st.session_state.player_freq['Proj Own'] = st.session_state.player_freq['Player'].map(player_own_dict)
                 st.session_state.player_freq = st.session_state.player_freq.set_index('Player')
         with col3:
             if proj_file is not None:
@@ -309,6 +353,7 @@ with tab2:
                                   split_portfolio['UTIL'].map(player_own_dict)])
                              
                         st.session_state.display_portfolio = split_portfolio[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'UTIL', 'Salary', 'Projection', 'Ownership']]
+                        st.session_state.export_portfolio = st.session_state.display_portfolio.replace(dk_ids, inplace=True)
                         hold_portfolio = display_portfolio.sort_values(by='Projection', ascending=False)
                         
                         st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:8].values, return_counts=True)),
@@ -326,9 +371,14 @@ with tab2:
         with col1:
             if 'display_portfolio' in st.session_state:
                 st.dataframe(st.session_state.display_portfolio.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
-
+                st.download_button(
+                    label="Export Full Frame",
+                    data=st.session_state.export_portfolio.to_csv().encode('utf-8'),
+                    file_name='portfolio_export.csv',
+                    mime='text/csv',
+                )
         with col2:
             if 'player_freq' in st.session_state:
-                st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+                st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(freq_format, precision=2), use_container_width = True)