Update audio_processor.py

audio_processor.py

@@ -2,6 +2,7 @@ import streamlit as st
 import speech_recognition as sr
 from gtts import gTTS
 import io
+import time
 
 def transcribe_audio(audio_file):
     recognizer = sr.Recognizer()
@@ -16,7 +17,7 @@ def transcribe_audio(audio_file):
         st.error(f"Transcription failed: {str(e)}")
         return ""
 
-def text_to_speech(text, target_lang):
+def text_to_speech(text, target_lang, max_retries=2):
     try:
         if not text:
             st.error("No text to convert to audio.")
@@ -27,16 +28,22 @@ def text_to_speech(text, target_lang):
         }
         lang_code = lang_map.get(target_lang, "en")
         st.write(f"Generating audio for {lang_code}")
-        tts = gTTS(text=text[:200], lang=lang_code, slow=False)  # Limit to 200 chars
-        audio_buffer = io.BytesIO()
-        tts.write_to_fp(audio_buffer)
-        audio_buffer.seek(0)
-        if audio_buffer.getbuffer().nbytes > 0:
-            st.success(f"Audio generated for {target_lang}")
-            return audio_buffer
-        else:
-            st.error("Audio buffer empty.")
-            return None
+        for attempt in range(max_retries):
+            try:
+                tts = gTTS(text=text[:200], lang=lang_code, slow=False)
+                audio_buffer = io.BytesIO()
+                tts.write_to_fp(audio_buffer)
+                audio_buffer.seek(0)
+                if audio_buffer.getbuffer().nbytes > 0:
+                    st.success(f"Audio generated for {target_lang} on attempt {attempt + 1}")
+                    return audio_buffer
+                time.sleep(2 ** attempt)  # Exponential backoff
+            except Exception as e:
+                st.warning(f"Attempt {attempt + 1} failed: {str(e)}")
+                if attempt == max_retries - 1:
+                    st.error(f"Audio failed after {max_retries} attempts. Try English/French.")
+                    return None
+        return None
     except Exception as e:
-        st.error(f"Audio failed: {str(e)}. Ensure target language is supported (e.g., English, French).")
+        st.error(f"Audio processing error: {str(e)}. Try English/French.")
         return None