Update piper_engine.py

piper_engine.py

@@ -1,39 +1,42 @@
-import subprocess
-import tempfile
-
-
-def piper_tts(
-    text,
-    model_path,
-    config_path,
-    speed=1.0,
-    pitch=1.0,
-):
-
-    # Output wav file
-    with tempfile.NamedTemporaryFile(
-        delete=False,
-        suffix=".wav"
-    ) as f:
-        out_path = f.name
-
-    cmd = [
-        "piper",
-        "--model", model_path,
-        "--config", config_path,
-        "--output_file", out_path,
-        "--length_scale", str(1 / speed),
-        "--pitch_scale", str(pitch),
-    ]
-
-    process = subprocess.Popen(
-        cmd,
-        stdin=subprocess.PIPE,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.PIPE,
-        text=True,
-    )
-
-    process.communicate(text)
-
-    return out_path
+# piper_engine.py
+
+import io
+import soundfile as sf
+
+from piper.voice import PiperVoice
+
+
+class PiperEngine:
+
+    def __init__(self, model_path):
+
+        self.voice = PiperVoice.load(model_path)
+
+    def synthesize(
+        self,
+        text,
+        speed=1.0,
+        pitch=1.0,
+        sample_rate=22050,
+    ):
+
+        # Generate raw audio
+        audio = self.voice.synthesize(
+            text,
+            length_scale=1 / speed,
+            pitch_scale=pitch,
+        )
+
+        # Convert to WAV buffer
+        buffer = io.BytesIO()
+
+        sf.write(
+            buffer,
+            audio,
+            sample_rate,
+            format="WAV",
+        )
+
+        buffer.seek(0)
+
+        return buffer