update

app.py

@@ -11,8 +11,8 @@ st.sidebar.title('Model Settings')
 # Define a function to manage state initialization
 def initialize_state():
     keys = ["model_submitted", "api_key", "endpoint_url", "deployment_name", "temperature", "max_tokens",
-            "data_processed", "group_name", "privilege_label", "protect_label", "num_run"]
-    defaults = [False, "", "", "", 0.5, 150, False, "", "", "", 1]
+            "data_processed", "group_name", "privilege_label", "protect_label", "num_run", "uploaded_file"]
+    defaults = [False, "", "", "", 0.5, 150, False, "", "", "", 1, None]
     for key, default in zip(keys, defaults):
         if key not in st.session_state:
             st.session_state[key] = default
@@ -43,17 +43,10 @@ if st.session_state.model_submitted:
     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)
-    uploaded_file = st.file_uploader("Choose a file")
+    st.session_state.uploaded_file = st.file_uploader("Choose a file")
 
-    if st.button("Reset Experiment Settings"):
-        st.session_state.group_name = ""
-        st.session_state.privilege_label = ""
-        st.session_state.protect_label = ""
-        st.session_state.num_run = 1
-        st.session_state.data_processed = False
-
-    if uploaded_file is not None and not st.session_state.data_processed:
-        data = StringIO(uploaded_file.getvalue().decode("utf-8"))
+    if st.session_state.uploaded_file is not None and not st.session_state.data_processed:
+        data = StringIO(st.session_state.uploaded_file.getvalue().decode("utf-8"))
         df = pd.read_csv(data)
 
         if st.button('Process Data'):
@@ -68,3 +61,11 @@ if st.session_state.model_submitted:
                 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 = ""
+        st.session_state.privilege_label = ""
+        st.session_state.protect_label = ""
+        st.session_state.num_run = 1
+        st.session_state.data_processed = False
+        st.session_state.uploaded_file = None