Update app.py

app.py

@@ -76,14 +76,15 @@ st.sidebar.button('Reset Chat', on_click=reset_conversation)
 # Add cautionary message about testing phase at the bottom of the sidebar
 st.sidebar.markdown("**Note**: This model is still in the testing phase. Responses may be inaccurate or undesired. Use it cautiously, especially for critical issues.")
 
-# Add logo to the sidebar
+# Add logo and text to the sidebar
 st.sidebar.image("https://assets.isu.pub/document-structure/221118065013-a6029cf3d563afaf9b946bb9497d45d4/v1/2841525b232adaef7bd0efe1da81a4c5.jpeg", width=200)
+st.sidebar.write("A product of Kisii University")
 
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
     st.session_state.message_count = 0
-    st.session_state.ask_recommendation = False
+    st.session_state.ask_intervention = False
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
@@ -99,12 +100,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 it's time to ask for recommendation
-    if st.session_state.message_count >= 10 and not st.session_state.ask_recommendation:
-        if st.button("Would you recommend this model to others?"):
-            st.session_state.ask_recommendation = True
-            # Add your logic here for handling the recommendation
-            # You may want to store the response in session state or a database
+    # Check if intervention is needed
+    if "suicide" in prompt.lower() or "help" in prompt.lower():
+        if not st.session_state.ask_intervention:
+            if st.button("Do you need to talk to a therapist?"):
+                st.session_state.ask_intervention = True
+                st.write("You can reach out to a certified therapist at +25493609747.")
 
     # Interact with the selected model
     assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])