Update app.py

app.py

@@ -10,6 +10,7 @@ def reset_conversation():
     Resets Conversation
     '''
     st.session_state.messages = []
+    st.session_state.user_interactions = 0  # Reset user interactions counter
     return None
 
 # Define model links for the Addiction Recovery and Mental Health models
@@ -28,8 +29,11 @@ model_pre_instructions = {
 def interact_with_together_api(messages, model_link):
     all_messages = []
 
-    # Add pre-instructions to the message history
-    all_messages.append({"role": "system", "content": model_pre_instructions[selected_model]})
+    # Add pre-instructions to the message history if it's the first interaction with this model
+    if not any("role" in msg for msg in messages):
+        all_messages.append({"role": "system", "content": model_pre_instructions[selected_model]})
+    else:
+        all_messages.append({"role": "system", "content": f"Switched to model: {selected_model}"})
 
     # Append user and assistant messages
     for human, assistant in messages:
@@ -70,15 +74,28 @@ def interact_with_together_api(messages, model_link):
 selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
-# Initialize chat history
+# Add cautionary message about testing phase at the bottom of the sidebar
+st.sidebar.markdown("**Note**: This model is still in the testing phase. Responses may be inaccurate or undesired. Use it cautiously, especially for critical issues.")
+
+# Initialize chat history and user interactions counter
 if "messages" not in st.session_state:
     st.session_state.messages = []
+if "user_interactions" not in st.session_state:
+    st.session_state.user_interactions = 0
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     with st.chat_message(message[0]):
         st.markdown(message[1])
 
+# Increment user interactions counter
+st.session_state.user_interactions += 1
+
+# Display recommendation popup after the first 10 user interactions
+if st.session_state.user_interactions == 10:
+    recommendation = st.sidebar.radio("Would you recommend this model to others?", ("Yes", "No"))
+    st.sidebar.write(f"You recommended: {recommendation}")
+
 # Accept user input
 if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     # Display user message in chat message container