update

pages/1_Injection.py

@@ -104,7 +104,7 @@ else:
 
                 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,st.session_state.proportion)
+                    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.proportion,st.session_state.prompt_template)
                     st.session_state.data_processed = True  # Mark as processed
 
                 st.write('Processed Data:', preprocessed_df)

util/injection.py

@@ -66,7 +66,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,proportion):
+def process_scores_multiple(df, num_run, parameters, privilege_label, protect_label, agent, group_name, occupation,proportion,template):
 
     print(f"Processing {len(df)} entries with {num_run} runs each.")
     """ Process entries and compute scores concurrently, with progress updates. """
@@ -77,7 +77,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
 
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
 
-                prompt_normal = create_summary(group_name, label, occupation,row,proportion)
+                prompt_normal = create_summary(group_name, label, occupation,row,proportion,template)
 
                 print(f"Run {run + 1} - Entry {index + 1} - {key}")
                 print("=============================================================")