update

pages/1_Injection.py

@@ -60,26 +60,28 @@ else:
 
     # Ensure experiment settings are only shown if model info is submitted
     if st.session_state.model_submitted:
-        df = None
+        df_raw = 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_raw = pd.read_csv("resume.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:
-            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)
-            df = df[df['Occupation'] == st.session_state.occupation]
+                df_raw = pd.read_csv(data)
+        if df_raw is not None:
+            if file_options == "Example":
+                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)
+                df = df_raw[df_raw['Occupation'] == st.session_state.occupation]
+            else:
+                df = df_raw
+
             st.write('Data:', df)
 
             # Button to add a new row