Update app.py

app.py

@@ -101,4 +101,20 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     st.session_state.message_count += 1
 
     # Check if intervention is needed based on bot response
-    if any(keyword
+    if any(keyword in prompt.lower() for keyword in ["human", "therapist"]):
+        if not st.session_state.ask_intervention:
+            if st.button("Do you need to talk to a therapist?"):
+                st.session_state.ask_intervention = True
+                st.write("You can reach out to a certified therapist at +25493609747.")
+
+    # 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():
+        st.markdown("AI is typing...")
+        st.empty()
+        st.markdown(assistant_response)
+
+    # Add assistant response to chat history
+    st.session_state.messages.append(("assistant", assistant_response))