Update app.py

app.py

@@ -28,10 +28,9 @@ model = SentenceTransformer("all-MiniLM-L6-v2").to(device)
 with open("technical_interviewer_prompt.txt", "r") as file:
     technical_interviewer_prompt = file.read()
 
-# Load prompts from files
 with open("question_generation_prompt.txt", "r") as file:
     question_generation_prompt = file.read()
-    
+
 st.title("Real-World Programming Question Mock Interview")
 
 # Initialize session state variables
@@ -107,41 +106,38 @@ if generate_button:
         f"\nPlease create a real-world interview question based on this information."
     )
     
-    # Generate response using GPT with detailed prompt (question generation prompt excluded here)
-    response = generate_response([{"role": "user", "content": detailed_prompt}])
+    # Generate response using GPT-4 with detailed prompt and debugging logs
+    response = generate_response([{"role": "user", "content": detailed_prompt}])  # Question generation prompt excluded here
 
-    # Store generated question in session state for persistence in sidebar and follow-up conversation state
+    # Store generated question in session state for persistence
     st.session_state.generated_question = response
 
-    # Add the generated question to the conversation history as an assistant message (omit prompts in the chat)
+    # Add the generated question and technical interviewer prompt as hidden start for follow-up
+    st.session_state.messages.append({"role": "user", "content": technical_interviewer_prompt})
     st.session_state.messages.append({"role": "assistant", "content": response})
 
     # Enable follow-up mode after generating the initial question
     st.session_state.follow_up_mode = True
 
 # Display chat messages from history on app rerun (for subsequent conversation)
-for message in st.session_state.messages:
-    if not message["content"].startswith("Please create a real-world interview question"):  # Exclude prompts
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
+for message in st.session_state.messages[2:]:  # Exclude the hidden messages (first two)
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 # Chatbox for subsequent conversations with assistant (follow-up mode)
 if st.session_state.follow_up_mode:
     if user_input := st.chat_input("Continue your conversation or ask follow-up questions here:"):
-        # Display user message in chat message container and add to session history
-        with st.chat_message("user"):
-            st.markdown(user_input)
-        
+        # Add the user's input to the session state
         st.session_state.messages.append({"role": "user", "content": user_input})
 
-        # Generate assistant's response based on follow-up input using technical_interviewer_prompt as system prompt
-        assistant_response = generate_response(
-            [{"role": "user", "content": user_input}]
-        )
+        # Generate assistant's response based on follow-up input
+        assistant_response = generate_response(st.session_state.messages)
 
+        # Display the assistant's response
         with st.chat_message("assistant"):
             st.markdown(assistant_response)
         
+        # Append the assistant's response to the conversation history
         st.session_state.messages.append({"role": "assistant", "content": assistant_response})
 
 st.sidebar.markdown("""
@@ -151,32 +147,15 @@ Enter a company name, topic, and level of difficulty, and it will transform a re
 Continue chatting with the AI interviewer in the chatbox.
 """)
 
-# Sidebar content to display persistent generated question (left sidebar)
+# Sidebar content to display persistent generated question
 st.sidebar.markdown("## Generated Question")
 if st.session_state.generated_question:
     st.sidebar.markdown(st.session_state.generated_question)
 else:
     st.sidebar.markdown("_No question generated yet._")
 
-# Right sidebar toggleable debug logs and code interpreter section
+# Right sidebar toggleable debug logs
 with st.expander("Debug Logs (Toggle On/Off)", expanded=False):
     if len(st.session_state.debug_logs) > 0:
         for log_entry in reversed(st.session_state.debug_logs):  # Show most recent logs first
             st.write(log_entry)
-
-st.sidebar.markdown("---")
-st.sidebar.markdown("## Python Code Interpreter")
-code_input = st.sidebar.text_area("Write your Python code here:")
-if st.sidebar.button("Run Code"):
-    try:
-        exec_globals = {}
-        exec(code_input, exec_globals)  # Execute user-provided code safely within its own scope.
-        output_key = [k for k in exec_globals.keys() if k != "__builtins__"]
-        if output_key:
-            output_value = exec_globals[output_key[0]]
-            st.sidebar.success(f"Output: {output_value}")
-        else:
-            st.sidebar.success("Code executed successfully!")
-            
-    except Exception as e:
-        st.sidebar.error(f"Error: {e}")