update

app.py

@@ -51,16 +51,16 @@ def invoke_retry(prompt, **parameters):
 
     raise Exception("Failed to complete the API call after maximum retry attempts.")
 
-def process_scores(df, num_run=1):
+def process_scores(df, num_run,parameters,privilege_label,protect_label):
     """ Process entries and compute scores concurrently, with progress updates. """
     scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
 
     for run in range(num_run):
         for index, row in tqdm(df.iterrows(), total=len(df), desc="Submitting Tasks"):
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, None]):
-                result = invoke_retry(create_summary(row,label), **parameters)
+                prompt_temp = create_summary(row,label)
+                result = invoke_retry(prompt_temp, **parameters)
                 scores[key][index].append(result)
-                break
 
     # Assign score lists and calculate average scores
     for category in ['Privilege', 'Protect', 'Neutral']:
@@ -163,7 +163,7 @@ if uploaded_file is not None:
 
 
         # Main Execution
-        df = process_scores(df,num_run)
+        df = process_scores(df,num_run,parameters,privilege_label,protect_label)
 
         # Display processed data
         st.write('Processed Data:', df)