Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from gradio_client import Client
 from st_audiorec import st_audiorec
 from gtts import gTTS
-import os
+from IPython.display import Audio, display
 
 # Constants
 TITLE = "AgriTure"
@@ -86,8 +86,12 @@ if wav_audio_data is not None:
     # Convert AI response to speech
     speech_file = text_to_speech(response)
 
+    # Display assistant response in chat message container
+    with st.chat_message("assistant", avatar='🌿'):
+        st.markdown(response)
+
     # Play the generated speech
-    st.audio(speech_file, format="audio/mp3")
+    display(Audio(speech_file, autoplay=True))
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
@@ -109,7 +113,7 @@ if prompt := textinput:
         st.markdown(response)
 
     # Play the generated speech
-    st.audio(speech_file, format="audio/mp3")
+    display(Audio(speech_file, autoplay=True))
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})