Update text_speech_utils.py

text_speech_utils.py

@@ -5,6 +5,7 @@ import soundfile as sf
 from gtts import gTTS
 import os
 import tempfile
+import time
 
 # Load Whisper model
 model = whisper.load_model("base")  # You can also try "small", "medium", or "large"
@@ -36,20 +37,32 @@ def transcribe_audio(filename):
 # Function to save text as an audio file using gTTS (Google Text-to-Speech)
 def save_text_as_audio(text, audio_filename):
     print("Converting text to speech...")
-    tts = gTTS(text=text, lang='en', slow=False)
     
-    # Save as mp3 file
-    mp3_filename = audio_filename.replace('.wav', '.mp3')
-    tts.save(mp3_filename)
-    print(f"Audio saved as {mp3_filename}")
+    try:
+        tts = gTTS(text=text, lang='en', slow=False)
+        
+        # Save as mp3 file
+        mp3_filename = audio_filename.replace('.wav', '.mp3')
+        tts.save(mp3_filename)
+        print(f"Audio saved as {mp3_filename}")
+        
+        # Convert mp3 to wav using pydub
+        audio = AudioSegment.from_mp3(mp3_filename)
+        audio.export(audio_filename, format="wav")
+        
+        # Delete the temporary mp3 file
+        os.remove(mp3_filename)
+        print(f"Audio converted and saved as {audio_filename}")
     
-    # Convert mp3 to wav using pydub
-    audio = AudioSegment.from_mp3(mp3_filename)
-    audio.export(audio_filename, format="wav")
-    
-    # Delete the temporary mp3 file
-    os.remove(mp3_filename)
-    print(f"Audio converted and saved as {audio_filename}")
+    except Exception as e:
+        print(f"Error occurred during text-to-speech conversion: {e}")
+        
+        # In case of error (like hitting the rate limit), wait and retry
+        print("Waiting for 60 seconds before retrying...")
+        time.sleep(60)  # wait for a minute before retrying
+        
+        # Retry the conversion
+        save_text_as_audio(text, audio_filename)  # Retry recursively
 
 # Function to play audio using pydub's playback
 def play_audio(filename):