Update app.py

app.py

@@ -83,6 +83,7 @@ st.sidebar.write("A product proudly developed by Kisii University")
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
+    st.session_state.message_count = 0
 
 # Keywords for intervention
 intervention_keywords = [
@@ -99,15 +100,23 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
         st.markdown(prompt)
     # Add user message to chat history
     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
-        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)
+        with st.chat_message("assistant"):
+            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)
 
     # 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.chat_message("assistant"):
-        st.markdown(assistant_response)
+    with st.empty():
+        st.markdown("AI is typing...")
+        st.empty()
+        with st.chat_message("assistant"):
+            st.markdown(assistant_response)
+
+    # Add assistant response to chat history
+    st.session_state.messages.append(("assistant", assistant_response))