Update app.py

app.py

@@ -3,6 +3,9 @@ from huggingface_hub import InferenceClient, HfApi
 import time
 import requests
 from requests.exceptions import RequestException
+from gtts import gTTS  # Google Text-to-Speech
+import tempfile
+import os
 
 # Set page config at the very beginning
 st.set_page_config(page_title="Phi-3.5 Chatbot", page_icon="🤖")
@@ -12,10 +15,7 @@ hf_token = st.text_input("Enter your Hugging Face API token", type="password")
 
 @st.cache_resource
 def get_client(token):
-    return InferenceClient(
-        "microsoft/Phi-3.5-mini-instruct",
-        token=token
-    )
+    return InferenceClient("microsoft/Phi-3.5-mini-instruct", token=token)
 
 def validate_token(token):
     try:
@@ -67,6 +67,13 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         st.error(f"An error occurred: {str(e)}")
         yield "I'm sorry, but I encountered an error while processing your request."
 
+def text_to_speech(text):
+    # Create a named temporary file
+    tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+    tts = gTTS(text=text, lang='en')
+    tts.save(tmp_file.name)
+    return tmp_file.name
+
 st.title("Phi-3.5 Mini Chatbot")
 
 if "messages" not in st.session_state:
@@ -99,4 +106,8 @@ if prompt := st.chat_input("What is your message?"):
             message_placeholder.markdown(response)
             full_response = response
 
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
+    st.session_state.messages.append({"role": "assistant", "content": full_response})
+
+    # Generate audio from the assistant's response
+    audio_file = text_to_speech(full_response)
+    st.audio(audio_file, format='audio/mp3')