'commit'

Util/ASRClient.py

@@ -0,0 +1,75 @@
+import asyncio
+from http import HTTPStatus
+from dashscope.audio.asr import Recognition
+import dashscope
+import logging
+from Config import Config
+
+# 初始化日志记录器
+logger = logging.getLogger(__name__)
+
+# 若没有配置环境变量，请用百炼API Key将下行替换为：dashscope.api_key = "sk-xxx"
+dashscope.api_key = Config.ALY_LLM_API_KEY
+
+class ASRClient:
+    """
+    阿里云语音识别客户端，用于处理语音文件的转写任务
+    使用 Recognition API 支持本地文件实时转写
+    """
+    
+    def __init__(self, api_key=None):
+        """
+        初始化ASR客户端
+        
+        Args:
+            api_key: 阿里云DashScope API密钥，若不提供则使用配置文件中的密钥
+        """
+        logger.info("开始初始化ASR客户端")
+        
+        try:
+            self.api_key = api_key or Config.ALY_LLM_API_KEY
+            dashscope.api_key = self.api_key
+            logger.info("ASR客户端初始化完成")
+        except Exception as e:
+            logger.error(f"初始化ASR客户端失败: {str(e)}", exc_info=True)
+            raise
+    
+    async def transcribe_file(self, file_path):
+        """
+        转写本地音频文件
+        
+        Args:
+            file_path: 本地音频文件路径
+            
+        Returns:
+            str: 转写后的文本，如果失败返回None
+        """
+        logger.info(f"开始转写文件: {file_path}")
+        
+        try:
+            recognition = Recognition(
+                model='paraformer-realtime-v1',
+                format='mp3',
+                sample_rate=16000,
+                callback=None
+            )
+            
+            # Run blocking call in executor
+            loop = asyncio.get_running_loop()
+            result = await loop.run_in_executor(None, lambda: recognition.call(file_path))
+            
+            if result.status_code == HTTPStatus.OK:
+                sentences = []
+                if 'sentence' in result.output:
+                    for s in result.output['sentence']:
+                        sentences.append(s['text'])
+                text = "".join(sentences)
+                logger.info("转写成功")
+                return text
+            else:
+                logger.error(f"转写失败: {result.code} - {result.message}")
+                return None
+                
+        except Exception as e:
+            logger.error(f"转写过程出错: {str(e)}", exc_info=True)
+            return None