Update app.py

Moved:
message = {"role": "assistant", "content": response}
st.session_state.messages.append(message)
from 109 to 121

app.py

@@ -106,19 +106,20 @@ def launch_bot():
                 # st.write("Mistral:" ) #Needs finishing
               #  else:
                 st.write(response) 
-        message = {"role": "assistant", "content": response}
-        st.session_state.messages.append(message)
 
     # If assistant has most recently reaponded create audio of response 
     # if st.session_state.messages[-1]["role"] == "assistant":
         audio_result = st.button("Convert to Audio 🔊") 
         if audio_result:
-            # with st.chat_message("assistant"):
-            st.write("Converting text to speech. This may take a few minutes.")            
+            with st.chat_message("assistant"):
+                st.write("Converting text to speech. This may take a few minutes.")            
             sound_file = BytesIO()
             tts = gTTS('Add text-to-speech to your app', lang='en')
             tts.write_to_fp(sound_file)
             st.audio(sound_file)
+
+        message = {"role": "assistant", "content": response}
+        st.session_state.messages.append(message)
                 
 if __name__ == "__main__":
     launch_bot()