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
-from IPython.display import Audio, display
+import os
 
 # Constants
 TITLE = "AgriTure"
@@ -86,12 +86,8 @@ 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
-    display(Audio(speech_file, autoplay=True))
+    st.audio(speech_file, format="audio/mp3")
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
@@ -113,7 +109,7 @@ if prompt := textinput:
         st.markdown(response)
 
     # Play the generated speech
-    display(Audio(speech_file, autoplay=True))
+    st.audio(speech_file, format="audio/mp3")
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})