Update app.py

app.py

@@ -1,14 +1,22 @@
-import os
 import uuid
 from pathlib import Path
 
-from fastapi import FastAPI, HTTPException
+import edge_tts
+from fastapi import FastAPI, HTTPException, Query
+from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import FileResponse, JSONResponse
 from pydantic import BaseModel
-import edge_tts
 
 app = FastAPI(title="Edge TTS API")
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=False,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 TMP_DIR = Path("/tmp/edge_tts_api")
 TMP_DIR.mkdir(parents=True, exist_ok=True)
 
@@ -21,6 +29,22 @@ class TTSRequest(BaseModel):
     pitch: str = "+0Hz"
 
 
+def simplify_voice(v: dict) -> dict:
+    short_name = v.get("ShortName", "")
+    locale = v.get("Locale", "")
+    return {
+        "Name": v.get("Name", ""),
+        "ShortName": short_name,
+        "FriendlyName": v.get("FriendlyName", ""),
+        "Locale": locale,
+        "Gender": v.get("Gender", ""),
+        "SuggestedCodec": v.get("SuggestedCodec", ""),
+        "Status": v.get("Status", ""),
+        "IsMultilingual": "Multilingual" in short_name,
+        "IsBulgarian": locale == "bg-BG",
+    }
+
+
 @app.get("/")
 async def root():
     return {
@@ -29,8 +53,8 @@ async def root():
         "routes": {
             "health": "/health",
             "voices": "/voices",
-            "tts": "/tts"
-        }
+            "tts": "/tts",
+        },
     }
 
 
@@ -40,15 +64,59 @@ async def health():
 
 
 @app.get("/voices")
-async def voices():
+async def voices(
+    locale: str | None = Query(default=None),
+    multilingual_only: bool = Query(default=False),
+    bulgarian_only: bool = Query(default=False),
+):
     try:
         voices_data = await edge_tts.list_voices()
-        voices_data = sorted(voices_data, key=lambda v: v.get("ShortName", ""))
-        return JSONResponse(content=voices_data)
+        simplified = [simplify_voice(v) for v in voices_data]
+        simplified.sort(key=lambda v: (v["Locale"], v["ShortName"]))
+
+        if locale:
+            simplified = [v for v in simplified if v["Locale"].lower() == locale.lower()]
+
+        if multilingual_only:
+            simplified = [v for v in simplified if v["IsMultilingual"]]
+
+        if bulgarian_only:
+            simplified = [v for v in simplified if v["IsBulgarian"]]
+
+        return JSONResponse(
+            content={
+                "count": len(simplified),
+                "voices": simplified,
+            }
+        )
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to fetch voices: {str(e)}")
 
 
+@app.get("/voices/bulgarian")
+async def voices_bulgarian():
+    try:
+        voices_data = await edge_tts.list_voices()
+        simplified = [simplify_voice(v) for v in voices_data]
+        simplified = [v for v in simplified if v["IsBulgarian"]]
+        simplified.sort(key=lambda v: v["ShortName"])
+        return {"count": len(simplified), "voices": simplified}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch Bulgarian voices: {str(e)}")
+
+
+@app.get("/voices/multilingual")
+async def voices_multilingual():
+    try:
+        voices_data = await edge_tts.list_voices()
+        simplified = [simplify_voice(v) for v in voices_data]
+        simplified = [v for v in simplified if v["IsMultilingual"]]
+        simplified.sort(key=lambda v: (v["Locale"], v["ShortName"]))
+        return {"count": len(simplified), "voices": simplified}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch multilingual voices: {str(e)}")
+
+
 @app.post("/tts")
 async def tts(req: TTSRequest):
     text = req.text.strip()
@@ -64,7 +132,7 @@ async def tts(req: TTSRequest):
             voice=req.voice,
             rate=req.rate,
             volume=req.volume,
-            pitch=req.pitch
+            pitch=req.pitch,
         )
         await communicate.save(str(output_file))
 
@@ -74,7 +142,7 @@ async def tts(req: TTSRequest):
         return FileResponse(
             path=str(output_file),
             media_type="audio/mpeg",
-            filename="speech.mp3"
+            filename="speech.mp3",
         )
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"TTS generation failed: {str(e)}")