Fix inference, setup global phonemizers and update requirements

app.py

@@ -1,24 +1,59 @@
 import gradio as gr
 import os
 from AinaTheme import theme
-from espeak_phonemizer import Phonemizer
+import phonemizer
 from dotenv import load_dotenv
+import logging
 
 load_dotenv()
 
 MAX_INPUT_TEXT_LEN = int(os.environ.get("MAX_INPUT_TEXT_LEN", default=325))
 
-#fonemitzador = Phonemizer("ca")
+critical_logger = logging.getLogger("phonemizer")
+critical_logger.setLevel(logging.CRITICAL)
+
+# init phonemizers
+global_phonemizer_central = phonemizer.backend.EspeakBackend(
+    language="ca",
+    preserve_punctuation=True,
+    with_stress=True,
+    language_switch="remove-flags",
+    logger=critical_logger,
+)
+
+global_phonemizer_valencia = phonemizer.backend.EspeakBackend(
+    language="ca-va",
+    preserve_punctuation=True,
+    with_stress=True,
+    language_switch="remove-flags",
+    logger=critical_logger,
+)
+
+global_phonemizer_occidental = phonemizer.backend.EspeakBackend(
+    language="ca-nw",
+    preserve_punctuation=True,
+    with_stress=True,
+    language_switch="remove-flags",
+    logger=critical_logger,
+)
+
+global_phonemizer_balear = phonemizer.backend.EspeakBackend(
+    language="ca-ba",
+    preserve_punctuation=True,
+    with_stress=True,
+    language_switch="remove-flags",
+    logger=critical_logger,
+)
 
 def phonemiser(text, dialect):
-    dialects = {"Central": "ca", "Valencian": "ca-va", "North-West": "ca-nw", "Balear": "ca-ba"}
-    dialect = dialects[dialect] #Define dialect from espeak-ng-data/lang/roa/ca*
-    fonemitzador = Phonemizer(dialect) #Set correct dialect for the phonemiser
-
+    phonemizers = {"Central": global_phonemizer_central,
+                "Valencian": global_phonemizer_valencia,
+                "North-West": global_phonemizer_occidental,
+                "Balear": global_phonemizer_balear}
+    
+    fonemitzador = phonemizers[dialect] #Set correct dialect for the phonemiser
     # synthesize
-
-    fonemes = fonemitzador.phonemize(text, keep_clause_breakers=True)
-
+    fonemes = fonemitzador.phonemize([text], strip=True, njobs=1)[0]
     return fonemes
 
 

requirements.txt

@@ -1,4 +1,4 @@
-aina-gradio-theme==2.0
-gradio==4.0.0
-espeak-phonemizer>=1.1.0,<2 
+aina-gradio-theme
+gradio
+phonemizer
 python-dotenv==1.0.0