Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 import os
 from dotenv import load_dotenv
 from requests.exceptions import RequestException
-import time
 
 load_dotenv()
 
@@ -77,10 +76,7 @@ def interact_with_together_api(messages, model_link):
         return None
 
 # Create sidebar with model selection dropdown and reset button
-selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()), key="model_selection")
-if st.session_state.get('selected_model') != selected_model:
-    st.session_state.selected_model = selected_model
-    st.info(f"You have switched to {selected_model}.")
+selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
 # Add cautionary message about testing phase at the bottom of the sidebar
@@ -110,28 +106,21 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
     st.session_state.messages.append(("user", prompt))
     st.session_state.message_count += 1
 
-    # Simulate typing
-    with st.empty():
-        st.markdown("AI is typing...")
-        time.sleep(0.5)
-        st.empty()
-        st.markdown("AI is typing...")
-        time.sleep(0.5)
-        st.empty()
-        st.markdown("AI is typing...")
-        time.sleep(0.5)
-
     # 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.chat_message("assistant"):
+        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.")
         
-        # 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 analyzing 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))