diff --git a/streamlit/interface_2.py b/streamlit/interface_2.py index 991baba..389c0d8 100644 --- a/streamlit/interface_2.py +++ b/streamlit/interface_2.py @@ -67,12 +67,12 @@ try: query = 'select * from "PROMPT_CONFIG"' cur.execute(query) fields = pd.DataFrame(cur.fetchall()) + fields.rename(columns={'FIELD_DESC': 'Field Name'}, inplace = True) fields.rename(columns={'PROMPT': 'Interrogation Question?'}, inplace = True) fields.rename(columns={'GROUP_ID': 'PRIORITY'}, inplace = True) fields.rename(columns={'FIELD_NAME': 'SF_DB_COL_NAME'}, inplace = True) fields.rename(columns={'FM_MODEL_ID': 'llm_selected'}, inplace = True) - error('table is empty') except Exception as e: st.write("Unable to fetch data from Snowflake: ",e) # fields = pd.read_csv('contract_fields.csv', encoding='unicode_escape', skipinitialspace=True)