load history and raw data from sf

This commit is contained in:
Mayank Aamseek
2024-03-27 13:58:40 +05:30
parent e87713bbd2
commit 9a25414e6d
+5 -2
View File
@@ -104,6 +104,7 @@ try:
history.rename(columns={'DATE_TIME': 'Date/Time'}, inplace = True)
history.rename(columns={'ACCURACY': 'Accuracy'}, inplace = True)
history.rename(columns={'ATTEMPT_NUM': 'Attempt #'}, inplace = True)
history = history[['Field Name', '# Contracts Tested', 'Username', 'Date/Time', 'Accuracy', 'Attempt #']]
except:
history = pd.read_csv('history.csv')
st.write("Local copy of TRAINING_ATTEMPT_LOGS table loaded")
@@ -516,8 +517,8 @@ if user_mail in user_list:
return df, history, attempt, raw_response_text
raw_response_text = ''
df = pd.DataFrame(columns=['Contract Name' ,'New Extracted value','Confidence Level','Snippet','New Page Number'
, 'Revised Prompt', 'Result'])
df = pd.DataFrame(columns=['Contract Name','Contract ID', 'Actual Value Stored','New Extracted value','Confidence Level'
,'Snippet','New Page Number', 'Revised Prompt', 'Result'])
if st.button("Test Configuration"):
df, history, attempt, raw_response_text = run_llm(attempt, bucket, contract_list, llm_selected, field, field_values, history)
@@ -538,6 +539,8 @@ if user_mail in user_list:
# df = pd.read_csv('results.csv')
# df['Result'] = df['Result'].astype('str')
# history = pd.read_csv('history.csv')
df = df[['Contract Name','Contract ID', 'Actual Value Stored','New Extracted value','Confidence Level'
,'Snippet', 'New Page Number', 'Revised Prompt', 'Result']]
st.dataframe(df)
st.dataframe(history)