Update app.py

app.py

@@ -1,47 +1,187 @@
-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 = []
+# st.session_state["openai_model"] = "gpt-4o"
 
 
-# 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')
+  
+# #     # 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...")
+     
+# # 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
+from streamlit_js_eval import streamlit_js_eval
+
+# 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
+
+def show_feedback():
+    st.session_state.feedback_shown = True
 
 # Setup stage for collecting user details
 if not st.session_state.setup_complete:
@@ -100,9 +240,9 @@ 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...")
-     
+
 # Interview phase
-if st.session_state.setup_complete and not st.session_state.chat_complete:
+if st.session_state.setup_complete and not st.session_state.feedback_shown and not st.session_state.chat_complete:
 
     st.info(
     """
@@ -114,12 +254,18 @@ if st.session_state.setup_complete and not st.session_state.chat_complete:
     # 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 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']}")
+            "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
@@ -140,7 +286,10 @@ 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)
@@ -151,4 +300,39 @@ 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
+
+# Show "Get Feedback" 
+if st.session_state.chat_complete and not st.session_state.feedback_shown:
+    if st.button("Get Feedback", on_click=show_feedback):
+        st.write("Fetching feedback...")
+
+# Show feedback screen
+if st.session_state.feedback_shown:
+    st.subheader("Feedback")
+
+    conversation_history = "\n".join([f"{msg['role']}: {msg['content']}" for msg in st.session_state.messages])
+
+    # Initialize new OpenAI client instance for feedback
+    feedback_client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
+
+    # Generate feedback using the stored messages and write a system prompt for the feedback
+    feedback_completion = feedback_client.chat.completions.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "system", "content": """You are a helpful tool that provides feedback on an interviewee performance.
+             Before the Feedback give a score of 1 to 10.
+             Follow this format:
+             Overal Score: //Your score
+             Feedback: //Here you put your feedback
+             Give only the feedback do not ask any additional questins.
+              """},
+            {"role": "user", "content": f"This is the interview you need to evaluate. Keep in mind that you are only a tool. And you shouldn't engage in any converstation: {conversation_history}"}
+        ]
+    )
+
+    st.write(feedback_completion.choices[0].message.content)
+
+    # Button to restart the interview
+    if st.button("Restart Interview", type="primary"):
+            streamlit_js_eval(js_expressions="parent.window.location.reload()")