From d98da03d3c5ca3f7afdd3f5df34fd9cac444add1 Mon Sep 17 00:00:00 2001 From: Mayank Aamseek Date: Mon, 8 Apr 2024 17:19:56 +0530 Subject: [PATCH] updated postprocessing --- streamlit/interface_3.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/streamlit/interface_3.py b/streamlit/interface_3.py index 997de1c..d5704de 100644 --- a/streamlit/interface_3.py +++ b/streamlit/interface_3.py @@ -550,8 +550,8 @@ if st.session_state.user_info['mail'] in user_list: df['Original Page Number'] = df['Original Page Number'].apply(lambda x: re.search(r'\d+', x).group( ) if isinstance(x, str) and re.search(r'\d+', x) is not None else " ") - df_date = df[df['SF_DB_COL_NAME'].str.contains('_DT')] - df_others = df[~df['SF_DB_COL_NAME'].str.contains('_DT')] + df_date = df[df['SF_DB_COL_NAME'].str.contains('_DT', na=False)] + df_others = df[~df['SF_DB_COL_NAME'].str.contains('_DT', na=False)] df_date['Actual Value Stored'] = pd.to_datetime(df_date['Actual Value Stored'],errors='coerce').dt.strftime('%Y-%m-%d').fillna(" ") df_date['New Extracted value'] = pd.to_datetime(df_date['New Extracted value'],errors='coerce').dt.strftime('%Y-%m-%d').fillna(" ") @@ -608,8 +608,12 @@ if st.session_state.user_info['mail'] in user_list: df = df[['Contract Name','Contract ID', 'SF_DB_COL_NAME', 'Actual Value Stored','New Extracted value','Confidence Level' ,'Snippet','Original Page Number', 'New Page Number', 'Revised Prompt', 'Result']] - df = pd.read_csv('results.csv') - df['Result'] = df['Result'].astype('str') + try: + df = pd.read_csv('results.csv') + df['Result'] = df['Result'].astype('str') + except: + df = pd.DataFrame(columns=['Contract Name' ,'New Extracted value','Confidence Level','Snippet','New Page Number' + , 'Revised Prompt', 'Result']) history = pd.read_csv('history.csv') st.dataframe(df) st.dataframe(history)