Update app.py

app.py

@@ -28,7 +28,6 @@ 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()
     
@@ -107,7 +106,7 @@ if generate_button:
         f"\nPlease create a real-world interview question based on this information."
     )
     
-    # Generate response using GPT-4 with detailed prompt and debugging logs
+    # Generate response using OpenAI API 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
@@ -130,11 +129,18 @@ if st.session_state.follow_up_mode:
         
         st.session_state.messages.append({"role": "user", "content": user_input})
 
-        # 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.
+        # Prepare messages to send to the assistant
+        # Include the technical interviewer prompt and generated question, but do not display them
+        # Add an instruction for the assistant to reply as a real-world interviewer would
+        assistant_instruction = (
+            "As a real-world interviewer, please reply to the candidate's follow-up questions "
+            "specific to the generated interview question, to the point, and in a natural, human-sounding way."
+        )
+
         messages_to_send = [
             {"role": "user", "content": technical_interviewer_prompt},
-            {"role": "assistant", "content": st.session_state.generated_question}
+            {"role": "assistant", "content": st.session_state.generated_question},
+            {"role": "user", "content": assistant_instruction}
         ] + st.session_state.messages
 
         assistant_response = generate_response(messages_to_send)
@@ -176,6 +182,5 @@ if st.sidebar.button("Run Code"):
             st.sidebar.success(f"Output: {output_value}")
         else:
             st.sidebar.success("Code executed successfully!")
-            
     except Exception as e:
         st.sidebar.error(f"Error: {e}")