Update app.py

app.py

@@ -11,7 +11,6 @@ def reset_conversation():
     Resets Conversation
     '''
     st.session_state.messages = []
-    st.session_state.ask_intervention = False
     return None
 
 def analyze_diagnosis(messages, model_link):
@@ -106,46 +105,35 @@ def interact_with_together_api(messages, model_link):
         st.error(f"Error communicating with the API: {e}")
         return None
 
-# Initialize chat history and session state attributes
+# Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
-    st.session_state.ask_intervention = False
 
 # Create sidebar with model selection dropdown and reset button
 selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
 reset_button = st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
 # Add diagnostic feature to the sidebar
-if st.session_state.messages:
-    with st.sidebar:
+if len(st.session_state.messages) >= 3:
+    with st.sidebar.beta_container():
         st.subheader("Diagnosis")
-        st.markdown("<div style='color: #1E90FF; font-size: 16px; font-weight: bold;'>Analyzing...</div>", unsafe_allow_html=True)
 
         diagnosis = analyze_diagnosis(st.session_state.messages, model_links[selected_model])
         if diagnosis:
-            st.markdown(f"<div style='color: #FF6347; font-size: 18px; font-weight: bold;'>Diagnosis: {diagnosis}</div>", unsafe_allow_html=True)
-
-# Add additional features on the sidebar
-st.sidebar.markdown("---")
-st.sidebar.subheader("Self-care and Support")
-st.sidebar.markdown("📝 Journaling")
-st.sidebar.markdown("🎯 Goal Setting")
-st.sidebar.markdown("😊 Mood Tracking")
-st.sidebar.markdown("🛀 Self-care Tips")
+            st.write(diagnosis)
 
 # Add cautionary message about testing phase at the bottom of the sidebar
-st.sidebar.markdown("---")
 st.sidebar.markdown("**Note**: This model is still in the beta phase. Responses may be inaccurate or undesired. Use it cautiously, especially for critical issues.")
 
 # Add logo and text to the sidebar
-st.sidebar.markdown("---")
 st.sidebar.image("https://assets.isu.pub/document-structure/221118065013-a6029cf3d563afaf9b946bb9497d45d4/v1/2841525b232adaef7bd0efe1da81a4c5.jpeg", width=200)
 st.sidebar.write("A product proudly developed by Kisii University")
 
 # Accept user input
 if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
     # Display user message in chat message container
-    st.write(prompt)
+    with st.chat_message("user"):
+        st.markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append(("user", prompt))
 
@@ -154,15 +142,17 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
 
     if assistant_response is not None:
         # Display assistant response in chat message container
-        st.write("AI is typing...")
-        st.write(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.")
+        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))