diff --git a/streamlit/interface_3.py b/streamlit/interface_3.py index ec49dca..842caab 100644 --- a/streamlit/interface_3.py +++ b/streamlit/interface_3.py @@ -51,28 +51,29 @@ except: user_mail = 'maamseek@aarete.com' try: - sf_secrets = get_secret() + sf_secrets = json.loads(get_secret()) conn = snowflake.connector.connect( - user=sf_secrets['user'], - password=sf_secrets['password'], + user=sf_secrets.get('user'), + password=sf_secrets.get('password'), account="aarete-doczyai", - role = sf_secrets['ROLE'], - warehouse=sf_secrets['warehouse'], - database=sf_secrets['database'], + role = "DEVADMIN", + warehouse="DEV_XS", + database="DOCZY_DEV", schema="STG" ) cur = conn.cursor() - query = 'select * from "TRAINING_DATA_RAW"' cur.execute(query) - field_values = pd.DataFrame(cur.fetchall()) - field_values['Contract ID'] = field_values['Document_Name'] - + field_values = pd.DataFrame.from_records(iter(cur), columns=[x[0] for x in cur.description]) + # st.write(field_values) + field_values['Document_Name'] = field_values['CONTRACT_TITLE'] + field_values['Contract ID'] = field_values['CONTRACT_TITLE'] + error('table values are incorrect') except: field_values = pd.read_csv('contract_field_values.csv', encoding='unicode_escape', skipinitialspace=True) field_values.rename(columns={'(internal) Document Name': 'Document_Name'}, inplace = True) field_values.rename(columns={'(Internal) Carveout ID': 'Contract ID'}, inplace = True) - st.write("conn failed") + # st.write("conn failed") try: query = 'select * from "PROMPT_CONFIG"'