'commit'

dsLightRag/Routes/ttsRoute.py

@@ -0,0 +1,145 @@
+import logging
+import uuid
+from typing import Optional
+
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+
+from Util.GengerateAudio import ByteDanceTTS
+
+# 创建声音生成路由
+router = APIRouter(prefix="/api/tts", tags=["声音生成"])
+
+# 配置日志
+logger = logging.getLogger(__name__)
+
+# 初始化TTS实例
+tts_instance = ByteDanceTTS()
+
+class TextToSpeechRequest(BaseModel):
+    """文本转语音请求参数"""
+    text: str
+    voice_type: Optional[str] = None
+    speed_ratio: Optional[float] = 1.0
+    volume_ratio: Optional[float] = 1.0
+    pitch_ratio: Optional[float] = 1.0
+    encoding: Optional[str] = "mp3"
+
+
+class TextToSpeechResponse(BaseModel):
+    """文本转语音响应"""
+    success: bool
+    message: str
+    audio_url: Optional[str] = None
+    audio_size: Optional[float] = None
+    audio_format: Optional[str] = None
+    request_id: Optional[str] = None
+
+
+@router.get("/voices/categories")
+async def get_voice_categories():
+    """
+    获取所有音色分类接口
+    返回所有可用的音色分类列表
+    """
+    try:
+        categories = tts_instance.get_all_categories()
+        return {
+            "success": True,
+            "data": categories,
+            "message": "获取音色分类成功"
+        }
+    except Exception as e:
+        logger.error(f"获取音色分类错误: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=f"获取音色分类失败: {str(e)}"
+        )
+
+
+@router.get("/voices/by-category/{category}")
+async def get_voices_by_category(category: str):
+    """
+    根据分类获取音色列表接口
+    Args:
+        category: 音色分类名称
+    返回指定分类下的所有音色列表
+    """
+    try:
+        voices = tts_instance.get_voices_by_category(category)
+        if not voices:
+            return {
+                "success": False,
+                "message": f"未找到分类 '{category}' 下的音色"
+            }
+            
+        return {
+            "success": True,
+            "data": voices,
+            "message": f"获取分类 '{category}' 下的音色成功"
+        }
+    except Exception as e:
+        logger.error(f"获取分类 '{category}' 下的音色错误: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=f"获取分类 '{category}' 下的音色失败: {str(e)}"
+        )
+
+
+@router.get("/voices/all")
+async def get_all_voices():
+    """
+    获取所有音色分类和音色列表接口
+    返回所有音色分类和每个分类下的音色列表
+    """
+    try:
+        all_voices = tts_instance.get_all_voices()
+        return {
+            "success": True,
+            "data": all_voices,
+            "message": "获取所有音色分类和列表成功"
+        }
+    except Exception as e:
+        logger.error(f"获取所有音色分类和列表错误: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=f"获取所有音色分类和列表失败: {str(e)}"
+        )
+
+
+@router.post("/generate", response_model=TextToSpeechResponse)
+async def generate_audio(request: TextToSpeechRequest):
+    """
+    文本转语音接口
+    根据输入文本和语音参数生成音频文件
+    """
+    try:
+        # 调用TTS工具生成音频
+        audio_url = tts_instance.generate_audio(
+            text=request.text,
+            voice_type=request.voice_type,
+            speed_ratio=request.speed_ratio,
+            volume_ratio=request.volume_ratio,
+            pitch_ratio=request.pitch_ratio,
+            encoding=request.encoding
+        )
+        
+        if audio_url:
+            return TextToSpeechResponse(
+                success=True,
+                message="音频生成成功",
+                audio_url=audio_url,
+                audio_format=request.encoding,
+                request_id=str(uuid.uuid4())
+            )
+        else:
+            return TextToSpeechResponse(
+                success=False,
+                message="音频生成失败"
+            )
+    except Exception as e:
+        logger.error(f"文本转语音接口错误: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=f"音频生成失败: {str(e)}"
+        )