Update app.py

app.py

@@ -39,9 +39,8 @@ def interact_with_together_api(messages, model_link):
         all_messages.append({"role": "user", "content": human})
         all_messages.append({"role": "assistant", "content": assistant})
 
-    # Add the latest user message if messages list is not empty
-    if messages:
-        all_messages.append({"role": "user", "content": messages[-1][1]})
+    # Add the latest user message
+    all_messages.append({"role": "user", "content": messages[-1][1]})
 
     url = "https://api.together.xyz/v1/chat/completions"
     payload = {
@@ -92,6 +91,14 @@ for message in st.session_state.messages:
     with st.chat_message(message[0]):
         st.markdown(message[1])
 
+# Keywords for intervention
+intervention_keywords = [
+    "human", "therapist", "someone", "died", "death", "help", "suicide", "suffering",
+    "crisis", "emergency", "support", "depressed", "anxiety", "lonely", "desperate",
+    "struggling", "counseling", "distressed", "hurt", "pain", "grief", "trauma", "die", "Kill",
+    "abuse", "danger", "risk", "urgent", "need assistance", "mental health", "talk to"
+]
+
 # 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
@@ -101,21 +108,21 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     st.session_state.messages.append(("user", prompt))
     st.session_state.message_count += 1
 
-# Check if intervention is needed based on user input or bot response
-if any(keyword in prompt.lower() for keyword in ["human", "therapist", "someone", "died", "death", "help", "suicide", "suffering", "crisis", "emergency", "support", "depressed", "anxiety", "lonely", "desperate", "struggling", "counseling", "distressed", "hurt", "pain", "grief", "trauma", "abuse", "danger", "risk", "urgent", "need assistance"] for prompt in st.session_state.messages):
-    # Intervention logic here
-    if not st.session_state.ask_intervention:
-        st.session_state.ask_intervention = True
-        st.markdown("<span style='color:red;'>It seems like you might need to talk to someone. Click [here](https://www.google.com) for immediate assistance.</span>", unsafe_allow_html=True)
+    # Check if intervention is needed based on bot response
+    if any(keyword in prompt.lower() for keyword in intervention_keywords):
+        # Intervention logic here
+        if not st.session_state.ask_intervention:
+            st.session_state.ask_intervention = True
+            st.markdown("<span style='color:red;'>It seems like you might need to talk to someone. Click [here](https://www.google.com) for immediate assistance.</span>", unsafe_allow_html=True)
 
-# Interact with the selected model
-assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])
+    # 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)
+    # 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))
+    # Add assistant response to chat history
+    st.session_state.messages.append(("assistant", assistant_response))