Update app.py

app.py

@@ -113,14 +113,12 @@ if "messages" not in st.session_state:
 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
+# Add diagnostic feature to the sidebar after at least 3 messages are input by the user
 if len(st.session_state.messages) >= 3:
-    with st.sidebar.beta_container():
-        st.subheader("Diagnosis")
-
-        diagnosis = analyze_diagnosis(st.session_state.messages, model_links[selected_model])
-        if diagnosis:
-            st.write(diagnosis)
+    diagnosis = analyze_diagnosis(st.session_state.messages, model_links[selected_model])
+    if diagnosis:
+        st.sidebar.subheader("Diagnosis")
+        st.sidebar.write(diagnosis)
 
 # Add cautionary message about testing phase at the bottom of the sidebar
 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.")
@@ -154,5 +152,5 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
                     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
+        # Add assistant response to chat histor
         st.session_state.messages.append(("assistant", assistant_response))