add chat box

app.py

@@ -106,8 +106,25 @@ for message in st.session_state.messages[1:]:  # Skip the system message
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
+# Chatbox for subsequent conversations with assistant
+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)
+    
+    st.session_state.messages.append({"role": "user", "content": user_input})
+
+    # Generate assistant's response based on follow-up input
+    assistant_response = generate_response(user_input)
+
+    with st.chat_message("assistant"):
+        st.markdown(assistant_response)
+    
+    st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+
 st.sidebar.markdown("""
 ## About
 This is a Real-World Interview Question Generator powered by OpenAI's API.
 Enter a company name, topic, and level of difficulty, and it will transform a relevant question into a real-world interview scenario!
+Continue chatting with the assistant in the chatbox below.
 """)