update

app.py

@@ -35,32 +35,33 @@ if st.sidebar.button("Reset Model Info"):
 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:
-    parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
-
-    st.session_state.group_name = st.text_input("Group Name", value=st.session_state.group_name)
-    st.session_state.privilege_label = st.text_input("Privilege Name", value=st.session_state.privilege_label)
-    st.session_state.protect_label = st.text_input("Protect Name", value=st.session_state.protect_label)
-    st.session_state.num_run = st.number_input("Number of runs", min_value=1, value=st.session_state.num_run)
+# File selection
+file_options = st.radio("Choose file source:", ["Upload", "Example"])
+if file_options == "Example":
+    example_data = """Put your CSV data here as a multi-line string or load from a file path."""
+    data = StringIO(example_data)
+    df = pd.read_csv(data)
+else:
     st.session_state.uploaded_file = st.file_uploader("Choose a file")
-
-    if st.session_state.uploaded_file is not None and not st.session_state.data_processed:
+    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 st.button('Process Data'):
-            # Initialize the correct agent based on model type
-            if model_type == 'AzureAgent':
-                agent = AzureAgent(st.session_state.api_key, st.session_state.endpoint_url, st.session_state.deployment_name)
-            else:
-                agent = GPTAgent(st.session_state.api_key, st.session_state.endpoint_url, st.session_state.deployment_name, api_version)
+# Ensure experiment settings are only shown if model info is submitted
+if st.session_state.model_submitted and df is not None:
+    if st.button('Process Data'):
+        # Initialize the correct agent based on model type
+        if model_type == 'AzureAgent':
+            agent = AzureAgent(st.session_state.api_key, st.session_state.endpoint_url, st.session_state.deployment_name)
+        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...'):
-                df = process_scores(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.data_processed = True  # Mark as processed
-            st.write('Processed Data:', df)
+        # Process data and display results
+        with st.spinner('Processing data...'):
+            parameters = {"temperature": st.session_state.temperature, "max_tokens": st.session_state.max_tokens}
+            df = process_scores(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.data_processed = True  # Mark as processed
+        st.write('Processed Data:', df)
 
     if st.button("Reset Experiment Settings"):
         st.session_state.group_name = ""

prompt_test.csv

@@ -0,0 +1,5 @@
+Age,MainBranch,Gender,EdLevel,YearsCode,YearsCodePro,Country,MentalHealth,Employment,HaveWorkedWith,PreviousSalary,ComputerSkills
+>35,Yes,Female,Undergraduate,6.0,5.0,Israel,No,1,JavaScript;Ruby;Homebrew;Yarn;React.js;Ruby on Rails;DigitalOcean;Heroku;PostgreSQL;Redis,138288.0,10
+>35,Yes,Female,Undergraduate,19.0,19.0,Ecuador,Yes,1,C#;SQL;ASP.NET;Microsoft SQL Server,30000.0,4
+<35,Yes,Female,Undergraduate,7.0,4.0,United Kingdom of Great Britain and Northern Ireland,No,1,Bash/Shell;Groovy;Java;Kotlin;Docker;Git;Kubernetes;Terraform;Spring;AWS;DynamoDB;PostgreSQL,93067.0,12
+>35,Yes,Female,Other,13.0,11.0,United States of America,Yes,1,C#;HTML/CSS;JavaScript;SQL;TypeScript;Docker;Git;Kubernetes;Angular;ASP.NET;ASP.NET Core ;Google Cloud Platform;Microsoft Azure;Elasticsearch;Microsoft SQL Server;SQLite,132500.0,16