update

app.py

@@ -35,36 +35,39 @@ if st.sidebar.button("Reset Model Info"):
 if st.sidebar.button("Submit Model Info"):
     st.session_state.model_submitted = True
 
-# File selection
-file_options = st.radio("Choose file source:", ["Upload", "Example"])
-if file_options == "Example":
-    df = pd.read_csv("prompt_test.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)
+
+
 
 # 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)
+if st.session_state.model_submitted:
+    df = None
+    file_options = st.radio("Choose file source:", ["Upload", "Example"])
+    if file_options == "Example":
+        df = pd.read_csv("prompt_test.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:
+        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...'):
-            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)
+            # 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 = ""
-        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
+        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