update

pages/1_Injection.py

@@ -106,7 +106,7 @@ else:
                 # Process data and display results
                 with st.spinner('Processing data...'):
                     parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
-                    preprocessed_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)
+                    preprocessed_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.charateristics)
                     st.session_state.data_processed = True  # Mark as processed
 
                 st.write('Processed Data:', preprocessed_df)

util/injection.py

@@ -69,7 +69,7 @@ def calculate_avg_score(score_list):
             avg_score = sum(valid_scores) / len(valid_scores)
             return avg_score
     return None
-def process_scores_multiple(df, num_run, parameters, privilege_label, protect_label, agent, group_name, occupation,charateristics=""):
+def process_scores_multiple(df, num_run, parameters, privilege_label, protect_label, agent, group_name, occupation,charateristics):
 
     print(f"Processing {len(df)} entries with {num_run} runs each.")
     """ Process entries and compute scores concurrently, with progress updates. """