Update app.py

app.py

@@ -184,16 +184,17 @@ with input_container:
 # Conditional display of AI generated responses as a function of user provided prompts
 if st.session_state['generated']:
     for i, generated_response in enumerate(st.session_state['generated']):
-                        # Display the user's message (if present)
-                        if i < len(st.session_state['past']):
-                            message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
-                        
-                        # Display the AI's generated response
-                        message(generated_response, key=str(i) + '_ai')
+        # Display the user's message (if present)
+        if i < len(st.session_state['past']):
+            message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
+        
+        # Display the AI's generated response
+        message(generated_response, key=str(i) + '_ai')
 else:
-                    # If no generated responses, display the initial message from the AI
-                    with st.chat_message("ai"):
-                        st.markdown("I'm your Mental health Assistant, How may I help you?")                                                              
+    # If no generated responses, display the initial message from the AI
+    if not st.session_state['past']:
+        st.markdown("I'm your Mental health Assistant, How may I help you?")
+                                                             
                             
 with st.sidebar.expander("Sentiment Analysis"):
         # Use the values stored in session state