Update app.py

app.py

@@ -3,10 +3,12 @@ from fastapi.responses import StreamingResponse
 from piper import PiperVoice
 import io
 import os
+import httpx
 import wave
 from pydantic import BaseModel
+from typing import Optional, Literal
 
-app = FastAPI()
+app = FastAPI("TTS App for my projects")
 
 # Path where models will be stored in the container
 MODEL_DIR = "./models"
@@ -40,6 +42,13 @@ VOICE_MAP = {
     "cy": {"gendered": False, "default": "cy_GB-gwryw_gogleddol-medium"}
 }
 
+IRISH_MAP = {
+    "Donegal": {"gendered":True, "male": "Dónall", "female":"Áine"},
+    "Kerry": {"gendered":True, "male": "Colm", "female":"Neasa"},
+    "Ring": {"gendered":False,"default":"Fianait"},
+    "Connemara": {"gendered":False,"default":"Sibéal"}
+}
+
 # Cache for loaded models to avoid re-loading from disk every request
 loaded_voices = {}
 
@@ -58,9 +67,10 @@ def get_voice(model_name: str):
 class TTSRequest(BaseModel):
     text: str
     language: str
-    gender: str
+    gender: Literal["male","female"] = "male"
+    dialect: Optional[Literal["Kerry", "Donegal", "Ring", "Connemara"]] = None
 
-@app.post("/tts")
+@app.post("/tts/piper")
 async def tts_post(request: TTSRequest):
     try:
         lang_code = request.language.lower()
@@ -98,7 +108,57 @@ async def tts_post(request: TTSRequest):
     except Exception as e:
         print(f"Error during TTS: {e}")
         raise HTTPException(status_code=500, detail=str(e))
+
+ABAIR_URL = "https://www.abair.ie/api2/synthesize"
+
+@app.post("/tts/irish")
+async def get_irish_tts(
+    request: TTSRequest
+):
+    """
+    Fetches Irish speech from ABAIR and returns an audio/wav stream.
+    """
+    if request.dialect:
+        dialect=request.dialect
+    else:
+        dialect="Donegal"
+    
+    if not IRISH_MAP.get(dialect, {}):
+        voice = "Áine"
+        dialect="Donegal"
+    elif IRISH_MAP[dialect]["gendered"]:
+        voice = IRISH_MAP[dialect][request.gender]
+    else:
+        voice =IRISH_MAP[dialect]["default"]
         
+    # Parameters for the ABAIR API
+    dialect_map = {
+        "Donegal": "ga_UL",
+        "Kerry": "ga_MU",
+        "Connemara": "ga_CO",
+        "Ring": "ga_MU"
+    }
+    params = {
+        "text": request.text,
+        "voice": voice,
+        "dialect": dialect_map.get(dialect, "ga_UL"),
+        "format": "wav",
+        "speed": 1.0
+    }
+
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.get(ABAIR_URL, params=params, timeout=10.0)
+            
+            if response.status_code != 200:
+                raise HTTPException(status_code=502, detail="ABAIR service error")
+
+            # Return the audio binary directly to the browser/client
+            return Response(content=response.content, media_type="audio/wav")
+
+        except httpx.RequestError as exc:
+            raise HTTPException(status_code=503, detail=f"Could not connect to ABAIR: {exc}")
+
 @app.get("/health")
 def home():
     return {"status": "Piper TTS is running"}
@@ -114,5 +174,5 @@ def home():
     return {
         "message": "Piper TTS API is running",
         "models_in_folder": files,
-        "supported_languages": [v for v in list(VOICE_MAP.keys())]
+        "supported_languages": [v for v in list(VOICE_MAP.keys())].append("ga")
     }