Update app.py

app.py

@@ -1,47 +1,46 @@
-# import streamlit as st
-# from openai import OpenAI
+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")
+# Setting up the Streamlit page configuration
+st.set_page_config(page_title="StreamlitChatMessageHistory", page_icon="💬")
+st.title("Chatbot")
 
-# # Initialize session state variables
-# st.session_state.setup_complete = False
-# st.session_state.user_message_count = 0
-# st.session_state.chat_complete = False
-# st.session_state.messages = []
-# st.session_state["openai_model"] = "gpt-4o"
+# Initialize session state variables
+st.session_state.setup_complete = False
+st.session_state.user_message_count = 0
+st.session_state.chat_complete = False
+st.session_state.messages = []
 
 
-# # Helper functions to update session state
-# def complete_setup():
-#     st.session_state.setup_complete = True
+# 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')
+# 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)
+    # 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')
+    # 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["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")
-# #     )
+    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...")
+    # 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:
@@ -101,141 +100,6 @@
 #     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"])
-
-#     # 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 that interviews {st.session_state['name']}. You should interview him for the "
-#                         f"{st.session_state['position']} position in 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
-
-
-
-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:
 
@@ -257,10 +121,8 @@ if st.session_state.setup_complete and not st.session_state.chat_complete:
     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']}")
+            "content": (f"You are an HR that interviews {st.session_state['name']}. You should interview him for the "
+                        f"{st.session_state['position']} position in the company {st.session_state['company']}")
         }]
 
     # Display chat messages
@@ -281,10 +143,7 @@ if st.session_state.setup_complete and not st.session_state.chat_complete:
                 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
-                        ],
+                        messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
                         stream=True,
                     )
                     response = st.write_stream(stream)
@@ -295,4 +154,152 @@ 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
+        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