Update annotator.py

annotator.py

@@ -18,18 +18,32 @@ import matplotlib.pyplot as plt
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
+service_creds = {
+    'speadsheet': st.secrets["gcp_service_account"][1:117],
+    'worksheet': st.secrets['gcp_service_account'][120:148],
+    'type': st.secrets['gcp_service_account'][151:166],
+    'project_id': st.secrets['gcp_service_account'][169:177],
+    'private_key_id': st.secrets['gcp_service_account'][180:220],
+    'private_key': st.secrets['gcp_service_account'][223:1955],
+    'client_email': st.secrets['gcp_service_account'][1958:2004],
+    'client_id': st.secrets['gcp_service_account'][2007:2028],
+    'auth_uri': st.secrets['gcp_service_account'][2031:2072],
+    'token_uri': st.secrets['gcp_service_account'][2075:2110],
+    'auth_provider_x509_cert_url': st.secrets['gcp_service_account'][2113:2155],
+    'client_x509_cert_url': st.secrets['gcp_service_account'][2158:2256]
+}
 
 @st.cache_resource
 def authorize_google_sheets():
     scope = ["https://www.googleapis.com/auth/spreadsheets", "https://www.googleapis.com/auth/drive"]
-    creds = service_account.Credentials.from_service_account_info(st.secrets["gcp_service_account"], scopes=scope)
+    creds = service_account.Credentials.from_service_account_info(service_creds, scopes=scope)
     client = gspread.authorize(creds)
     return client
 
 
-def get_google_sheet_data():
+def get_google_sheet_data(rec_name):
     client = authorize_google_sheets()
-    sheet = client.open("XP_final_annotations").worksheet("rec_name")
+    sheet = client.open("XP_final_annotations").worksheet(rec_name)
     data = sheet.get_all_records()
     df = pd.DataFrame(data)
     return df
@@ -333,7 +347,7 @@ def iden():
                     st.header("Annotated DataFrame")
                     st.write(
                         ":orange[Feel free to also access the dataframe on google sheet [link](https://docs.google.com/spreadsheets/d/119CGzxLv0kclMMb3SDYYwrULn2WY77OqDrzR6McEYO0/edit?gid=0#gid=0)]")
-                    df = get_google_sheet_data()
+                    df = get_google_sheet_data(rec_name)
                     df_display = df.astype(str)
                     st.write(df_display)
                     st.markdown('#####')