used BUSINESS_CONFIG table

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)