update

pages/1_Injection.py

@@ -60,27 +60,27 @@ else:
 
     # Ensure experiment settings are only shown if model info is submitted
     if st.session_state.model_submitted:
-        df_raw = 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 = None
         file_options = st.radio("Choose file source:", ["Upload", "Example"])
         if file_options == "Example":
             #df = pd.read_csv("prompt_test.csv")
-            df_raw = pd.read_csv("resume.csv")
+            df = pd.read_csv("resume.csv")
+            df = df[df['Occupation'] == st.session_state.occupation]
         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_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
+                df= pd.read_csv(data)
+        if df is not None:
 
             st.write('Data:', df)