Update app.py

app.py

@@ -12,11 +12,20 @@ def reset_conversation():
     st.session_state.messages = []
     return None
 
-# Define model link for the Nous model
-model_link = "NousResearch/Nous-Hermes-2-Yi-34B"
-
-# Function to interact with the Nous model via the Together API
-def interact_with_together_api(messages):
+# Define model links for the Addiction Recovery and Mental Health models
+model_links = {
+    "Addiction recovery AI": "NousResearch/Nous-Hermes-2-Yi-34B",
+    "Mental health AI": "NousResearch/Nous-Hermes-2-Yi-34B"
+}
+
+# Define pre-instructions for each model
+model_pre_instructions = {
+    "Addiction recovery AI": "This AI is trained to provide support and guidance related to addiction recovery.",
+    "Mental health AI": "This AI is trained to provide support and guidance related to mental health."
+}
+
+# Function to interact with the selected model via the Together API
+def interact_with_together_api(messages, model_link):
     all_messages = []
 
     if not messages:  # If history is empty
@@ -56,7 +65,9 @@ def interact_with_together_api(messages):
 
     return assistant_response
 
-# Create sidebar with reset button
+# Create sidebar with model selection dropdown and reset button
+selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
+st.sidebar.write(f"Pre-Instructions: {model_pre_instructions[selected_model]}")
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
 # Initialize chat history
@@ -69,15 +80,15 @@ for message in st.session_state.messages:
         st.markdown(message[1])
 
 # Accept user input
-if prompt := st.chat_input("Hi, I'm the Nous model, ask me a question"):
+if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append(("user", prompt))
 
-    # Interact with the Nous model
-    assistant_response = interact_with_together_api(st.session_state.messages)
+    # Interact with the selected model
+    assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])
 
     # Display assistant response in chat message container
     with st.empty():