Update tts_engine.py

tts_engine.py

@@ -1,15 +1,33 @@
 import os
 from TTS.api import TTS
 
+# Map our app languages → Coqui TTS codes
+LANG_MAP = {
+    "english": "en",
+    "yoruba": "en",   # no native Yoruba model, fallback to English phonemes
+    "hausa": "en",    # same here, fallback
+    "igbo": "en",     # same here
+    "pidgin": "en",   # close enough to English
+    "esan": "en",
+    "tiv": "en",
+    "calabar": "en",
+    "benin": "en",
+    "french": "fr-fr",
+    "portuguese": "pt-br"
+}
+
 class TTSEngine:
     def __init__(self, use_coqui=False):
         self.use_coqui = use_coqui
         self.tts = None
 
         if self.use_coqui:
-            # Multilingual model (handles Yoruba/Hausa/Igbo tones)
-            self.tts = TTS("tts_models/multilingual/multi-dataset/your_tts",
-                           progress_bar=False, gpu=False)
+            # Multilingual model (limited codes available)
+            self.tts = TTS(
+                "tts_models/multilingual/multi-dataset/your_tts",
+                progress_bar=False,
+                gpu=False
+            )
 
     def speak(self, text, lang="english", voice_clone=False):
         """
@@ -21,21 +39,24 @@ class TTSEngine:
         out_file = "output.wav"
 
         if self.use_coqui:
+            # normalize language
+            lang_code = LANG_MAP.get(lang.lower(), "en")
+
             if voice_clone and os.path.exists("my_voice.wav"):
                 self.tts.tts_to_file(
                     text=text,
                     file_path=out_file,
                     speaker_wav="my_voice.wav",
-                    language=lang
+                    language=lang_code
                 )
             else:
                 self.tts.tts_to_file(
                     text=text,
                     file_path=out_file,
-                    language=lang
+                    language=lang_code
                 )
         else:
-            # Lazy import to avoid crash on Hugging Face
+            # Lazy import to avoid HF crash
             import pyttsx3
             engine = pyttsx3.init()
             engine.save_to_file(text, out_file)