Update app.py

app.py

@@ -128,16 +128,29 @@ st.sidebar.image("https://assets.isu.pub/document-structure/221118065013-a6029cf
 st.sidebar.write("A product proudly developed by Kisii University")
 
 # Accept user input
-if prompt := st.text_input(f"Hi, I'm {selected_model}, let's chat"):
+if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
     # 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 selected model
     assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])
 
     if assistant_response is not None:
         # Display assistant response in chat message container
+        with st.empty():
+            st.markdown("AI is typing...")
+            st.empty()
+            st.markdown(assistant_response)
+
+            # Check if intervention is needed based on 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"]):
+                # Intervention logic here
+                if not st.session_state.ask_intervention:
+                    if st.button("After the analysing our session you may need some extra help, so you can reach out to a certified therapist at +25493609747 Name: Ogega feel free to talk"):
+                        st.write("You can reach out to a certified therapist at +25493609747.")
+
+        # Add assistant response to chat history
         st.session_state.messages.append(("assistant", assistant_response))
-
-# Display chat messages from history
-    # Add assistant response to chat history
-    st.session_state.messages.append(("assistant", assistant_response))