Update app.py

app.py

@@ -47,75 +47,10 @@ if not st.session_state.setup_complete:
     # 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=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')
-
-#     # 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(st.session_state["company"])
-#     )
-
-
-
-#     # Button to complete setup
-#     if st.button("Start Interview", on_click=complete_setup):
-#         st.write("Setup complete. Starting interview...")
-     
 # Interview phase
 if st.session_state.setup_complete and not st.session_state.chat_complete:
 
-    st.info(
-    """
-    Start by introducing yourself
-    """,
-    icon="👋",
-    )
-
     # Initialize OpenAI client
     client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
 
@@ -160,152 +95,4 @@ if st.session_state.setup_complete and not st.session_state.chat_complete:
 
     # Check if the user message count reaches 5
     if st.session_state.user_message_count >= 5:
-        st.session_state.chat_complete = True
-
-
-
-
-
-
-
-# import streamlit as st
-# from openai import OpenAI
-
-# # Setting up the Streamlit page configuration
-# st.set_page_config(page_title="StreamlitChatMessageHistory", page_icon="💬")
-# st.title("Chatbot")
-
-# # Initialize session state variables
-# if "setup_complete" not in st.session_state:
-#     st.session_state.setup_complete = False
-# if "user_message_count" not in st.session_state:
-#     st.session_state.user_message_count = 0
-# if "feedback_shown" not in st.session_state:
-#     st.session_state.feedback_shown = False
-# if "chat_complete" not in st.session_state:
-#     st.session_state.chat_complete = False
-# if "messages" not in st.session_state:
-#     st.session_state.messages = []
-
-# # Helper functions to update session state
-# 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')
-
-#     # 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=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')
-
-#     # 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(st.session_state["company"])
-#     )
-
-
-
-#     # Button to complete setup
-#     if st.button("Start Interview", on_click=complete_setup):
-#         st.write("Setup complete. Starting interview...")
-
-# # Interview phase
-# if st.session_state.setup_complete and not st.session_state.chat_complete:
-
-#     st.info(
-#     """
-#     Start by introducing yourself
-#     """,
-#     icon="👋",
-#     )
-
-#     # Initialize OpenAI client
-#     client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
-
-#     # Setting OpenAI model if not already initialized
-#     if "openai_model" not in st.session_state:
-#         st.session_state["openai_model"] = "gpt-4o"
-
-#     # Initializing the system prompt for the chatbot
-#     if not st.session_state.messages:
-#         st.session_state.messages = [{
-#             "role": "system",
-#             "content": (f"You are an HR executive that interviews an interviewee called {st.session_state['name']} "
-#                         f"with experience {st.session_state['experience']} and skills {st.session_state['skills']}. "
-#                         f"You should interview him for the position {st.session_state['level']} {st.session_state['position']} "
-#                         f"at the company {st.session_state['company']}")
-#         }]
-
-#     # Display chat messages
-#     for message in st.session_state.messages:
-#         if message["role"] != "system":
-#             with st.chat_message(message["role"]):
-#                 st.markdown(message["content"])
-
-#     # Handle user input and OpenAI response
-#     # Put a max_chars limit
-#     if st.session_state.user_message_count < 5:
-#         if prompt := st.chat_input("Your response", max_chars=1000):
-#             st.session_state.messages.append({"role": "user", "content": prompt})
-#             with st.chat_message("user"):
-#                 st.markdown(prompt)
-
-#             if st.session_state.user_message_count < 4:
-#                 with st.chat_message("assistant"):
-#                     stream = client.chat.completions.create(
-#                         model=st.session_state["openai_model"],
-#                         messages=[
-#                             {"role": m["role"], "content": m["content"]}
-#                             for m in st.session_state.messages
-#                         ],
-#                         stream=True,
-#                     )
-#                     response = st.write_stream(stream)
-#                 st.session_state.messages.append({"role": "assistant", "content": response})
-
-#             # Increment the user message count
-#             st.session_state.user_message_count += 1
-
-#     # Check if the user message count reaches 5
-#     if st.session_state.user_message_count >= 5:
-#         st.session_state.chat_complete = True
+        st.session_state.chat_complete = True