Update app.py

app.py

@@ -11,22 +11,32 @@ DetectorFactory.seed = 0
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 def generate_response(message, system_message, max_tokens, temperature, top_p):
+    if not message:
+        return "Please enter a message.", None
+
     messages = [{"role": "system", "content": system_message}]
     messages.append({"role": "user", "content": message})
 
     response = ""
-    for message in client.chat_completion(
+    for msg in client.chat_completion(
         messages,
         max_tokens=max_tokens,
         stream=True,
         temperature=temperature,
         top_p=top_p,
     ):
-        token = message.choices[0].delta.content
+        token = msg.choices[0].delta.content
         response += token
 
+    # Detect language and set TTS language
+    try:
+        lang = detect(message)
+    except Exception as e:
+        st.error(f"Error detecting language: {e}")
+        lang = 'en'
+
     # Generate speech from the response text using gTTS
-    tts = gTTS(text=response, lang='en' if detect(message) == 'en' else 'ar')
+    tts = gTTS(text=response, lang='ar' if lang == 'ar' else 'en')
     audio_file = "response.mp3"
     tts.save(audio_file)
     return response, audio_file
@@ -48,17 +58,18 @@ if st.button("Generate Response"):
     response, audio_file = generate_response(message, system_message, max_tokens, temperature, top_p)
     st.write(response)
 
-    # Get base64 encoded audio data
-    audio_base64 = get_base64_audio(audio_file)
-    
-    # HTML to autoplay the audio
-    audio_html = f"""
-    <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)
+    if audio_file:
+        # Get base64 encoded audio data
+        audio_base64 = get_base64_audio(audio_file)
+        
+        # HTML to autoplay the audio
+        audio_html = f"""
+        <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)
+        # Clean up the audio file
+        os.remove(audio_file)