update

pages/1_Injection.py

@@ -73,15 +73,17 @@ else:
         file_options = st.radio("Choose file source:", ["Upload", "Example"])
         if file_options == "Example":
             #df = pd.read_csv("prompt_test.csv")
-            df_raw = pd.read_csv("resume.csv")
-            df = df_raw[df_raw['Occupation'] == st.session_state.occupation]
+            df = pd.read_csv("resume.csv")
         else:
             st.session_state.uploaded_file = st.file_uploader("Choose a file")
             if st.session_state.uploaded_file is not None:
                 data = StringIO(st.session_state.uploaded_file.getvalue().decode("utf-8"))
                 df = pd.read_csv(data)
         if df is not None:
-            st.write('Data:', df)
+            if file_options == "Example":
+                st.write('Data:',df[df["Occupation"] == st.session_state.occupation])
+            else:
+                st.write('Data:', df)
 
             # Button to add a new row
 
@@ -107,7 +109,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,st.session_state.additional_charateristics)
+                    preprocessed_df = process_scores_multiple(df[df["Occupation"] == st.session_state.occupation], 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.additional_charateristics)
                     st.session_state.data_processed = True  # Mark as processed
 
                 st.write('Processed Data:', preprocessed_df)