Update app.py

app.py

@@ -179,13 +179,12 @@ with input_container:
                 speech_fp = text_to_speech(response)
                 # Play the speech
                 st.audio(speech_fp, format='audio/mp3')
-
-## Conditional display of AI generated responses as a function of user provided prompts
-with response_container:        
-    if st.session_state['generated']:
-        for i in range(len(st.session_state['generated'])):
-            message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
-            message(st.session_state["generated"][i], key=str(i))
+                
+                # 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']):
+                        message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
+                        message(generated_response, key=str(i) + '_ai')
             
 with st.sidebar.expander("Sentiment Analysis"):
         # Use the values stored in session state