Update app.py

app.py

@@ -28,6 +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
@@ -87,7 +91,9 @@ if generate_button:
     # Clear session state and start fresh with follow-up mode disabled
     st.session_state.messages = []
     st.session_state.follow_up_mode = False
-    
+
+    st.session_state.messages.append([{"role": "user", "content": question_generation_prompt}])
+
     # Create a query from user inputs and find the most relevant question
     query = f"{company} {difficulty} {topic}"
     top_question = find_top_question(query)
@@ -132,7 +138,7 @@ if st.session_state.follow_up_mode:
         # 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
+            [{"role": "user", "content": technical_interviewer_prompt}] + st.session_state.messages
         )
 
         with st.chat_message("assistant"):