Update app.py

app.py

@@ -109,9 +109,12 @@ if generate_button:
     # Generate response using GPT-4 with detailed prompt and debugging logs
     response = generate_response([{"role": "assistant", "content": question_generation_prompt}, {"role": "user", "content": detailed_prompt}])
 
-    # Store generated question in session state for persistence in sidebar
+    # 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 to the conversation history as an assistant message (to make it part of follow-up conversations)
+    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
 
@@ -129,7 +132,8 @@ 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 as system prompt
+        # Generate assistant's response based on follow-up input using technical_interviewer_prompt as system prompt,
+        # including the generated question in context.
         assistant_response = generate_response(
             [{"role": "assistant", "content": technical_interviewer_prompt}] + st.session_state.messages
         )