12.26

services/a2f_api/api.py

@@ -22,6 +22,7 @@ class TextRequest(BaseModel):
     split_punctuations: str = None
     max_sentence_length: int = None
     first_sentence_split_size: int = None
+    tts_provider: str = 'pyttsx3'  # 'pyttsx3' 或 'edge-tts'
 
 @app.get('/health')
 async def health():
@@ -30,7 +31,10 @@ async def health():
 @app.post('/text-to-blendshapes')
 async def text_to_blendshapes(request: TextRequest):
     try:
-        service = TextToBlendShapesService(lang=request.language)
+        service = TextToBlendShapesService(
+            lang=request.language,
+            tts_provider=request.tts_provider
+        )
         result = service.text_to_blend_shapes(
             request.text,
             segment=request.segment,
@@ -46,7 +50,10 @@ async def text_to_blendshapes(request: TextRequest):
 @app.post('/text-to-blendshapes/stream')
 async def text_to_blendshapes_stream(request: TextRequest):
     async def generate():
-        service = TextToBlendShapesService(lang=request.language)
+        service = TextToBlendShapesService(
+            lang=request.language,
+            tts_provider=request.tts_provider
+        )
         try:
             for message in service.iter_text_to_blend_shapes_stream(
                 request.text,