update

pages/1_Injection.py

@@ -5,7 +5,6 @@ from util.injection import process_scores_multiple
 from util.model import AzureAgent, GPTAgent
 import os
 
-# Set up the Streamlit interface
 st.title('Result Generation')
 
 def check_password():
@@ -73,10 +72,6 @@ else:
 
         if df is not None:
 
-            # 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"]
             categories = list(df["Occupation"].unique())
 
             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)
@@ -104,7 +99,6 @@ else:
                     agent = GPTAgent(st.session_state.api_key, st.session_state.endpoint_url,
                                      st.session_state.deployment_name, api_version)
 
-                # 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.charateristics)