Update app.py

app.py

@@ -17,28 +17,86 @@ st.session_state["openai_model"] = "gpt-4o"
 def complete_setup():
     st.session_state.setup_complete = True
 
+# # Setup stage for collecting user details
+# if not st.session_state.setup_complete:
+#     st.subheader('Personal Information')
+  
+#     # Get personal information input
+#     st.session_state["name"] = st.text_input(label="Name", value="",  placeholder="Enter your name", max_chars=40)
+    
+#     # Company and Position Section
+#     st.subheader('Company and Position')
+
+#     st.session_state["position"] = st.selectbox(
+#         "Choose a position",
+#         ("Data Scientist", "Data Engineer", "ML Engineer", "BI Analyst", "Financial Analyst"),
+#         index=("Data Scientist", "Data Engineer", "ML Engineer", "BI Analyst", "Financial Analyst").index("Data Scientist")
+#     )
+
+#     st.session_state["company"] = st.selectbox(
+#         "Select a Company",
+#         ("Amazon", "Meta", "Udemy", "365 Company", "Nestle", "LinkedIn", "Spotify"),
+#         index=("Amazon", "Meta", "Udemy", "365 Company", "Nestle", "LinkedIn", "Spotify").index("Amazon")
+#     )
+
+#     # Button to complete setup
+#     if st.button("Start Interview", on_click=complete_setup):
+#         st.write("Setup complete. Starting interview...")
+
 # Setup stage for collecting user details
 if not st.session_state.setup_complete:
     st.subheader('Personal Information')
-  
+
+    # Initialize session state for personal information
+    if "name" not in st.session_state:
+        st.session_state["name"] = ""
+    if "experience" not in st.session_state:
+        st.session_state["experience"] = ""
+    if "skills" not in st.session_state:
+        st.session_state["skills"] = ""
+
+   
     # Get personal information input
-    st.session_state["name"] = st.text_input(label="Name", value="",  placeholder="Enter your name", max_chars=40)
+    st.session_state["name"] = st.text_input(label="Name", value=st.session_state["name"], placeholder="Enter your name", max_chars=40)
+    st.session_state["experience"] = st.text_area(label="Experience", value=st.session_state["experience"], placeholder="Describe your experience", max_chars=200)
+    st.session_state["skills"] = st.text_area(label="Skills", value=st.session_state["skills"], placeholder="List your skills", max_chars=200)
+
     
     # Company and Position Section
     st.subheader('Company and Position')
 
-    st.session_state["position"] = st.selectbox(
-        "Choose a position",
-        ("Data Scientist", "Data Engineer", "ML Engineer", "BI Analyst", "Financial Analyst"),
-        index=("Data Scientist", "Data Engineer", "ML Engineer", "BI Analyst", "Financial Analyst").index("Data Scientist")
-    )
+    # Initialize session state for company and position information and setting default values 
+    if "level" not in st.session_state:
+        st.session_state["level"] = "Junior"
+    if "position" not in st.session_state:
+        st.session_state["position"] = "Data Scientist"
+    if "company" not in st.session_state:
+        st.session_state["company"] = "Amazon"
+
+    col1, col2 = st.columns(2)
+    with col1:
+        st.session_state["level"] = st.radio(
+            "Choose level",
+            key="visibility",
+            options=["Junior", "Mid-level", "Senior"],
+            index=["Junior", "Mid-level", "Senior"].index(st.session_state["level"])
+        )
+
+    with col2:
+        st.session_state["position"] = st.selectbox(
+            "Choose a position",
+            ("Data Scientist", "Data Engineer", "ML Engineer", "BI Analyst", "Financial Analyst"),
+            index=("Data Scientist", "Data Engineer", "ML Engineer", "BI Analyst", "Financial Analyst").index(st.session_state["position"])
+        )
 
     st.session_state["company"] = st.selectbox(
         "Select a Company",
         ("Amazon", "Meta", "Udemy", "365 Company", "Nestle", "LinkedIn", "Spotify"),
-        index=("Amazon", "Meta", "Udemy", "365 Company", "Nestle", "LinkedIn", "Spotify").index("Amazon")
+        index=("Amazon", "Meta", "Udemy", "365 Company", "Nestle", "LinkedIn", "Spotify").index(st.session_state["company"])
     )
 
+
+
     # Button to complete setup
     if st.button("Start Interview", on_click=complete_setup):
         st.write("Setup complete. Starting interview...")