Update app.py

app.py

@@ -3,6 +3,10 @@ from huggingface_hub import InferenceClient
 from gtts import gTTS
 import base64
 import os
+from langdetect import detect, DetectorFactory
+
+# To ensure consistent results from langdetect
+DetectorFactory.seed = 0
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
@@ -22,7 +26,7 @@ def generate_response(message, system_message, max_tokens, temperature, top_p):
         response += token
 
     # Generate speech from the response text using gTTS
-    tts = gTTS(text=response, lang='en')
+    tts = gTTS(text=response, lang='en' if detect(message) == 'en' else 'ar')
     audio_file = "response.mp3"
     tts.save(audio_file)
     return response, audio_file
@@ -32,7 +36,7 @@ def get_base64_audio(audio_file):
         audio_data = f.read()
     return base64.b64encode(audio_data).decode()
 
-st.title("Chatbot with Voice Response")
+st.title("A Chatbot with Voice Response")
 
 message = st.text_input("Enter your message:")
 system_message = st.text_input("System message:", value="You are a friendly Chatbot.")