Update app.py

app.py

@@ -84,12 +84,18 @@ st.sidebar.write("A product proudly developed by Kisii University")
 if "messages" not in st.session_state:
     st.session_state.messages = []
     st.session_state.message_count = 0
+    st.session_state.ask_intervention = False
+
+# 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])
 
 # 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",
+    "struggling", "counseling", "distressed", "hurt", "pain", "grief", "trauma", "die", "Kill",
     "abuse", "danger", "risk", "urgent", "need assistance", "mental health", "talk to"
 ]
 
@@ -102,11 +108,12 @@ 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 bot response
-    if any(keyword in prompt.lower() for keyword in intervention_keywords):
-        # Intervention logic here
-        with st.chat_message("assistant"):
+    # Check for intervention keywords in user input
+    for keyword in intervention_keywords:
+        if keyword in prompt.lower():
+            # Intervention logic here
             st.markdown("<span style='color:red;'>I have a feeling you may need to talk to a therapist. If you agree with me please contact +254793609747; Name: Davis. If you dont then keep talking to me as we figure this out.</span>", unsafe_allow_html=True)
+            break  # Exit loop once intervention is triggered
 
     # Interact with the selected model
     assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])
@@ -115,8 +122,7 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     with st.empty():
         st.markdown("AI is typing...")
         st.empty()
-        with st.chat_message("assistant"):
-            st.markdown(assistant_response)
+        st.markdown(assistant_response)
 
     # Add assistant response to chat history
     st.session_state.messages.append(("assistant", assistant_response))