update

pages/1_Injection.py

@@ -26,9 +26,9 @@ def check_password():
 def initialize_state():
     keys = ["model_submitted", "api_key", "endpoint_url", "deployment_name", "temperature", "max_tokens",
             "data_processed", "group_name", "occupation", "privilege_label", "protect_label", "num_run",
-            "uploaded_file", "additional_charateristics"]
+            "uploaded_file", "additional_charateristics", "occupation_submitted"]
     defaults = [False, "", "https://safeguard-monitor.openai.azure.com/", "gpt35-1106", 0.0, 150, False, "Gender",
-                "Programmer", "Male", "Female", 1, None, None]
+                "Programmer", "Male", "Female", 1, None, None, False]
     for key, default in zip(keys, defaults):
         if key not in st.session_state:
             st.session_state[key] = default
@@ -58,32 +58,33 @@ else:
     if st.sidebar.button("Submit Model Info"):
         st.session_state.model_submitted = True
 
-    # Ensure experiment settings are only shown if model info is submitted
-    if st.session_state.model_submitted:
+    categories = ["HR", "DESIGNER", "INFORMATION-TECHNOLOGY", "TEACHER", "ADVOCATE", "BUSINESS-DEVELOPMENT",
+                  "HEALTHCARE", "FITNESS", "AGRICULTURE", "BPO", "SALES", "CONSULTANT", "DIGITAL-MEDIA",
+                  "AUTOMOBILE", "CHEF", "FINANCE", "APPAREL", "ENGINEERING", "ACCOUNTANT", "CONSTRUCTION",
+                  "PUBLIC-RELATIONS", "BANKING", "ARTS", "AVIATION"]
+
+    st.session_state.occupation = st.selectbox("Occupation", options=categories, index=categories.index(
+        st.session_state.occupation) if st.session_state.occupation in categories else 0)
 
-        categories = ["HR", "DESIGNER", "INFORMATION-TECHNOLOGY", "TEACHER", "ADVOCATE", "BUSINESS-DEVELOPMENT",
-                      "HEALTHCARE", "FITNESS", "AGRICULTURE", "BPO", "SALES", "CONSULTANT", "DIGITAL-MEDIA",
-                      "AUTOMOBILE", "CHEF", "FINANCE", "APPAREL", "ENGINEERING", "ACCOUNTANT", "CONSTRUCTION",
-                      "PUBLIC-RELATIONS", "BANKING", "ARTS", "AVIATION"]
+    if st.button("Submit Occupation Selection"):
+        st.session_state.occupation_submitted = True
 
-        st.session_state.occupation = st.selectbox("Occupation", options=categories, index=categories.index(
-            st.session_state.occupation) if st.session_state.occupation in categories else 0)
+    # Ensure experiment settings are only shown if model info is submitted
+    if st.session_state.model_submitted and st.session_state.occupation_submitted:
 
         df = None
         file_options = st.radio("Choose file source:", ["Upload", "Example"])
         if file_options == "Example":
             #df = pd.read_csv("prompt_test.csv")
             df = pd.read_csv("resume.csv")
+            df = df[df["Occupation"] == st.session_state.occupation]
         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:
-            if file_options == "Example":
-                st.write('Data:',df[df["Occupation"] == st.session_state.occupation])
-            else:
-                st.write('Data:', df)
+            st.write('Data:', df)
 
             # Button to add a new row
 
@@ -109,7 +110,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[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)
+                    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)
                     st.session_state.data_processed = True  # Mark as processed
 
                 st.write('Processed Data:', preprocessed_df)