Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from huggingface_hub import InferenceClient
 from gtts import gTTS
+import base64
 import os
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -26,25 +27,34 @@ def generate_response(message, system_message, max_tokens, temperature, top_p):
     tts.save(audio_file)
     return response, audio_file
 
-st.title("Chatbot with Voice Response")
+def get_base64_audio(audio_file):
+    with open(audio_file, "rb") as f:
+        audio_data = f.read()
+    return base64.b64encode(audio_data).decode()
 
-message = st.text_input("Enter your message:")
-submit = st.button("Generate Response")
+st.title("Hugging Face Chatbot with Voice Response")
 
-if submit and message:
-    system_message = st.text_input("System message:", value="You are a friendly Chatbot.")
-    max_tokens = st.slider("Max new tokens", 1, 2048, 512)
-    temperature = st.slider("Temperature", 0.1, 4.0, 0.7, step=0.1)
-    top_p = st.slider("Top-p (nucleus sampling)", 0.1, 1.0, 0.95, step=0.05)
+message = st.text_input("Enter your message:")
+system_message = st.text_input("System message:", value="You are a friendly Chatbot.")
+max_tokens = st.slider("Max new tokens", 1, 2048, 512)
+temperature = st.slider("Temperature", 0.1, 4.0, 0.7, step=0.1)
+top_p = st.slider("Top-p (nucleus sampling)", 0.1, 1.0, 0.95, step=0.05)
 
+if st.button("Generate Response"):
     response, audio_file = generate_response(message, system_message, max_tokens, temperature, top_p)
     st.write(response)
 
-    # Add HTML to autoplay the audio
+    # Get base64 encoded audio data
+    audio_base64 = get_base64_audio(audio_file)
+    
+    # HTML to autoplay the audio
     audio_html = f"""
-    <audio autoplay>
-    <source src="data:audio/mpeg;base64,{open(audio_file, "rb").read().encode("base64").decode()}" type="audio/mp3">
+    <audio controls autoplay>
+    <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
     Your browser does not support the audio element.
     </audio>
     """
     st.markdown(audio_html, unsafe_allow_html=True)
+
+    # Clean up the audio file
+    os.remove(audio_file)