Update app.py

app.py

@@ -28,9 +28,10 @@ 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
@@ -109,35 +110,38 @@ if generate_button:
     # 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
+    # Store generated question in session state for persistence in sidebar and follow-up conversation state
     st.session_state.generated_question = response
 
-    # 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[2:]:  # Exclude the hidden messages (first two)
+for message in st.session_state.messages:
     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:"):
-        # Add the user's input to the session state
+        # Display user message in chat message container and add to session history
+        with st.chat_message("user"):
+            st.markdown(user_input)
+        
         st.session_state.messages.append({"role": "user", "content": user_input})
 
-        # Generate assistant's response based on follow-up input
-        assistant_response = generate_response(st.session_state.messages)
+        # Generate assistant's response based on follow-up input using technical_interviewer_prompt and generated question,
+        # but do not include them in the displayed conversation history.
+        messages_to_send = [
+            {"role": "system", "content": technical_interviewer_prompt},
+            {"role": "assistant", "content": st.session_state.generated_question}
+        ] + st.session_state.messages
+
+        assistant_response = generate_response(messages_to_send)
 
-        # 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("""
@@ -147,15 +151,32 @@ 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
+# Sidebar content to display persistent generated question (left sidebar)
 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
+# Right sidebar toggleable debug logs and code interpreter section
 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}")