updated postprocessing

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)