From 574c16d1fa9bc54d06af8e69d95cdc6cb4e7279e Mon Sep 17 00:00:00 2001 From: Mayank Aamseek Date: Tue, 26 Mar 2024 15:24:40 +0530 Subject: [PATCH] used BUSINESS_CONFIG table --- streamlit/interface_3.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/streamlit/interface_3.py b/streamlit/interface_3.py index 429ec03..d30debc 100644 --- a/streamlit/interface_3.py +++ b/streamlit/interface_3.py @@ -43,7 +43,7 @@ with st.sidebar: # st.write("Doczy") try: - util.setup_page(REDIRECT_URI) + # util.setup_page(REDIRECT_URI) _,c1= st.columns([4,1]) c1.write(f"User: **{st.session_state.user_info['displayName']}**") user_mail = st.session_state.user_info['mail'] @@ -76,19 +76,21 @@ except: # st.write("conn failed") try: - query = 'select * from "PROMPT_CONFIG"' + query = 'select * from "BUSINESS_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={'FIELD_NAME': 'Field Name'}, inplace = True) + fields.rename(columns={'QUESTION': 'Interrogation Question?'}, inplace = True) fields.rename(columns={'GROUP_ID': 'PRIORITY'}, inplace = True) fields.rename(columns={'FIELD_NAME': 'SF_DB_COL_NAME'}, inplace = True) + fields['Field Name'] = fields['SF_DB_COL_NAME'] fields.rename(columns={'FM_MODEL_ID': 'llm_selected'}, inplace = True) - error('table is empty') + # error('table is empty') except: fields = pd.read_csv('contract_fields.csv', encoding='unicode_escape', skipinitialspace=True) fields = fields.drop_duplicates(subset='Field Name', keep="first").sort_values('Field Name') fields = fields[~fields['Field Name'].isnull()] + st.write("conn failed") if user_mail in user_list: @@ -185,7 +187,10 @@ if user_mail in user_list: column_name = fields.loc[fields['Field Name'] == field, 'SF_DB_COL_NAME'].iloc[0] field_values = field_values[~field_values[column_name].isnull()] # to be deleted later + # df = pd.DataFrame({'col':contract_list}) + # st.write(df) contract_list = [contract for contract in contract_list if contract.rsplit('/',1)[1].replace(' MU','').replace('_MU','').replace('.txt','') in list(field_values['Document_Name'])] + if contract_count == 'All': contract_count = len(contract_list)