update

pages/2_Injection_Multiple.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 from io import StringIO
-from util.generation import process_scores
+from util.generation import process_scores_multiple
 from util.model import AzureAgent, GPTAgent
 
 # Set up the Streamlit interface
@@ -74,7 +74,7 @@ if st.session_state.model_submitted:
             # Process data and display results
             with st.spinner('Processing data...'):
                 parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
-                df = process_scores(df, st.session_state.num_run, parameters, [st.session_state.privilege_label,st.session_state.protect_label], agent, st.session_state.group_name,st.session_state.occupation, test_type='multiple')
+                df = process_scores_multiple(df, st.session_state.num_run, parameters, st.session_state.privilege_label,st.session_state.protect_label, agent, st.session_state.group_name,st.session_state.occupation)
                 st.session_state.data_processed = True  # Mark as processed
 
             st.write('Processed Data:', df)

pages/4_Injection_Single.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 from io import StringIO
-from util.generation import process_scores
+from util.generation import process_scores_single
 from util.model import AzureAgent, GPTAgent
 
 # Set up the Streamlit interface
@@ -73,9 +73,9 @@ if st.session_state.model_submitted:
             # Process data and display results
             with st.spinner('Processing data...'):
                 parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
-                df = process_scores(df, st.session_state.num_run, parameters, [st.session_state.counterfactual_label],
+                df = process_scores_single(df, st.session_state.num_run, parameters, st.session_state.counterfactual_label,
                                     agent, st.session_state.group_name,
-                                    st.session_state.occupation,test_type='multiple')
+                                    st.session_state.occupation)
                 st.session_state.data_processed = True  # Mark as processed
 
             st.write('Processed Data:', df)