update

util/injection.py

@@ -69,8 +69,8 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
     while attempts < max_attempts:
         try:
             score_text = agent.invoke(prompt, **parameters)
-            # print(f"Score text: {score_text}")
-            # print("=============================================================")
+            print(f"Score text: {score_text}")
+            print("=============================================================")
             if string_input:
                 return score_text
             score = re.search(r'\d+', score_text)
@@ -115,8 +115,8 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
                 prompt_charateristics = create_summary(group_name, label, occupation, charateristics)
                 prompt_normal = create_summary(group_name, label, occupation, '\n'.join(summary))
 
-                # print(f"Run {run + 1} - Entry {index + 1} - {key}")
-                # print("=============================================================")
+                print(f"Run {run + 1} - Entry {index + 1} - {key}")
+                print("=============================================================")
                 result_charateristics = invoke_retry(prompt_charateristics, agent, parameters)
                 result_normal = invoke_retry(prompt_normal, agent, parameters)
                 scores[key+"_characteristics"][index].append(result_charateristics)
@@ -127,17 +127,12 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
     # Ensure all scores are lists and calculate average scores
     for category in ['Privilege_characteristics', 'Privilege_normal', 'Protect_characteristics', 'Protect_normal',
                      'Neutral_characteristics', 'Neutral_normal']:
-        # Debug: Print the scores for the current category
-        print(f"Processing category: {category}")
-        print(f"Scores: {scores[category]}")
+
 
         # Ensure the scores are lists and check before assignment
         series_data = [lst if isinstance(lst, list) else [lst] for lst in scores[category]]
         df[f'{category}_Scores'] = series_data
 
-        # Debug: Check the Series after assignment
-        print(f"Series for {category}_Scores:\n{df[f'{category}_Scores']}")
-
         # Calculate the average score with additional debug info
         def calculate_avg_score(score_list):
             if isinstance(score_list, list) and score_list:
@@ -150,7 +145,6 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
 
         df[f'{category}_Avg_Score'] = df[f'{category}_Scores'].apply(calculate_avg_score)
 
-        # Debug: Print the calculated average scores
-        print(f"Average scores for {category}:\n{df[f'{category}_Avg_Score']}")
+
 
     return df