Update app.py

app.py

@@ -68,14 +68,19 @@ def get_speakers_dict():
         logger.error(f"Failed to import from new settings structure: {e}")
         # Fallback to old structure if new one fails
         try:
-            from maliba_ai.config.speakers import Adame, Moussa, Bourama, Modibo, Seydou
+            from maliba_ai.config.speakers import Adame, Moussa, Bourama, Modibo, Seydou, Amadou, Bakary, Ngolo, Ibrahima, Amara
             logger.info("Using fallback old speaker structure")
             return {
                 "Adama": Adame,
                 "Moussa": Moussa, 
                 "Bourama": Bourama,
                 "Modibo": Modibo,
-                "Seydou": Seydou
+                "Seydou": Seydou,
+                "Amadou": Amadou,
+                "Bakary": Bakary,
+                "Ngolo": Ngolo,
+                "Ibrahima": Ibrahima,
+                "Amara": Amara
             }
         except Exception as e2:
             logger.error(f"Failed to import speakers: {e2}")
@@ -104,7 +109,7 @@ def initialize_model_once():
         start_time = time.time()
         
         # Use the new import structure from the README
-        from maliba_ai.tts.inference import BambaraTTSInference
+        from maliba_ai.tts import BambaraTTSInference
         
         model = BambaraTTSInference()
         speakers = get_speakers_dict()
@@ -247,8 +252,8 @@ examples = [
     ["To tɔ nantan ni lafiya, o ka fisa ni so fa dumuniba kɛlɛma ye.", "Amadou"],  # Warm and friendly voice for wisdom saying
     ["Mali ye jamana ɲuman ye!", "Bakary"],  # Deep, authoritative tone for patriotic statement
     ["An ka ɲɔgɔn dɛmɛ ka baara kɛ ɲɔgɔn fɛ", "Ngolo"],  # Youthful and energetic for collaboration
-    ["Hakili bɛ yɔrɔ min na, sabali bɛ yen", "Ibrahima"],  # Calm and measured for philosophical thought
-    ["Aw ni ce. Seidu bɛ aw fo wa aw ka yafa a ma, ka da a kan tuma dɔw la kow ka can.", "Amara"],  # Melodic and smooth for poetic expression
+    ["Hakili to yɔrɔ min na, sabali bɛ yen", "Ibrahima"],  # Calm and measured for philosophical thought
+    ["Dɔnko ɲuman ye, a bɛ dɔn mɔgɔ kɔnɔ", "Amara"],  # Melodic and smooth for poetic expression
 ]
 
 def get_safe_examples():
@@ -444,7 +449,8 @@ def build_interface():
 def main():
     """Main function to launch the Gradio interface"""
     logger.info("Starting Bambara TTS Gradio interface.")
-
+    
+    # DO NOT preload - let it initialize on first request only (like your working version)
     interface = build_interface()
     interface.launch(
         server_name="0.0.0.0",