Update tts_engine.py

tts_engine.py

@@ -1,31 +1,44 @@
-import os
-import pyttsx3
-from TTS.api import TTS
-
-class TTSEngine:
-    def __init__(self, use_coqui=False):
-        self.use_coqui = use_coqui
-        if self.use_coqui:
-            # Multilingual model (supports Yoruba/Hausa/Igbo accents)
-            self.tts = TTS("tts_models/multilingual/multi-dataset/your_tts", progress_bar=False, gpu=False)
-
-    def speak(self, text, lang="english", voice_clone=False):
-        """
-        Returns path to generated audio file (for Gradio playback).
-        """
-        if not text:
-            return None
-
-        out_file = "output.wav"
-
-        if self.use_coqui:
-            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)
-            else:
-                self.tts.tts_to_file(text=text, file_path=out_file, language=lang)
-        else:
-            engine = pyttsx3.init()
-            engine.save_to_file(text, out_file)
-            engine.runAndWait()
-
-        return out_file
+import os
+from TTS.api import TTS
+
+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)
+
+    def speak(self, text, lang="english", voice_clone=False):
+        """
+        Returns path to generated audio file (for Gradio playback).
+        """
+        if not text:
+            return None
+
+        out_file = "output.wav"
+
+        if self.use_coqui:
+            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
+                )
+            else:
+                self.tts.tts_to_file(
+                    text=text,
+                    file_path=out_file,
+                    language=lang
+                )
+        else:
+            # Lazy import to avoid crash on Hugging Face
+            import pyttsx3
+            engine = pyttsx3.init()
+            engine.save_to_file(text, out_file)
+            engine.runAndWait()
+
+        return out_file