initial commit

test_asr.py

@@ -0,0 +1,53 @@
+import asyncio
+import logging
+import wave
+from custom_agent import SenseVoiceSTT
+from livekit import rtc
+from livekit.agents import utils
+
+# 设置日志级别以查看输出
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("test-asr")
+
+async def test():
+    # 替换为你本地的一个音频文件路径
+    audio_path = "/home/verachen/Music/voice/2food.wav" 
+    
+    # 初始化 ASR
+    stt = SenseVoiceSTT(url="http://10.6.80.21:5003/asr-blackbox")
+    
+    print(f"Testing ASR connectivity with file: {audio_path}")
+    
+    try:
+        # 读取音频文件
+        with wave.open(audio_path, 'rb') as wf:
+            frames = wf.readframes(wf.getnframes())
+            # 简单构造一个 AudioBuffer (假设是单声道 16kHz)
+            # 实际上 SenseVoiceSTT._recognize_impl 会用 combine_audio_frames(buffer).to_wav_bytes()
+            # 所以我们需要传递一个包含 AudioFrame 的 list
+            
+            # 这里我们模拟一个 Frame
+            frame = rtc.AudioFrame(
+                data=frames,
+                sample_rate=wf.getframerate(),
+                num_channels=wf.getnchannels(),
+                samples_per_channel=wf.getnframes()
+            )
+            
+            # 调用 recognize
+            result = await stt.recognize(buffer=[frame])
+            
+            if result.alternatives:
+                print(f"\n--- ASR Result ---")
+                print(f"Text: {result.alternatives[0].text}")
+                print(f"------------------\n")
+            else:
+                print("ASR returned no text.")
+                
+    except FileNotFoundError:
+        print(f"Error: Audio file not found at {audio_path}")
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
+if __name__ == "__main__":
+    asyncio.run(test())