Create app.py

app.py

@@ -0,0 +1,49 @@
+from fastapi import FastAPI, Response
+from fastapi.responses import StreamingResponse
+from piper import PiperVoice
+import io
+import os
+
+app = FastAPI()
+
+# Path where models will be stored in the container
+MODEL_DIR = "./models"
+os.makedirs(MODEL_DIR, exist_ok=True)
+
+# Cache for loaded models to avoid re-loading from disk every request
+loaded_voices = {}
+
+def get_voice(model_name: str):
+    if model_name not in loaded_voices:
+        # Assumes model files (onnx and json) are in MODEL_DIR
+        model_path = os.path.join(MODEL_DIR, f"{model_name}.onnx")
+        config_path = f"{model_path}.json"
+        
+        if not os.path.exists(model_path):
+            raise FileNotFoundError(f"Model {model_name} not found.")
+            
+        loaded_voices[model_name] = PiperVoice.load(model_path, config_path)
+    return loaded_voices[model_name]
+
+@app.get("/tts")
+async def tts(text: str, model: str = "en_US-lessac-medium"):
+    try:
+        voice = get_voice(model)
+        
+        # Create an in-memory buffer for the WAV file
+        wav_buffer = io.BytesIO()
+        with io.BytesIO() as f:
+            voice.synthesize(text, f)
+            audio_data = f.getvalue()
+            
+        return Response(content=audio_data, media_type="audio/wav")
+    except Exception as e:
+        return {"error": str(e)}
+
+@app.get("/health")
+def home():
+    return {"status": "Piper TTS is running"}
+
+@app.get("/")
+def home():
+    return {"status": "Piper TTS is running"}