sf conn reverted

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"'