From b0887f17678e0e5c4bd895e36695b242f2f1aee8 Mon Sep 17 00:00:00 2001
From: mengzhe.cmz <mengzhe.cmz@alibaba-inc.com>
Date: 星期四, 23 三月 2023 19:59:28 +0800
Subject: [PATCH] Merge branch 'dev_gzf' of github.com:alibaba-damo-academy/FunASR into dev_gzf

---
 funasr/runtime/python/websocket/ASR_client.py |   95 +++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 95 insertions(+), 0 deletions(-)

diff --git a/funasr/runtime/python/websocket/ASR_client.py b/funasr/runtime/python/websocket/ASR_client.py
new file mode 100644
index 0000000..8010b18
--- /dev/null
+++ b/funasr/runtime/python/websocket/ASR_client.py
@@ -0,0 +1,95 @@
+import pyaudio
+# import websocket #鍖哄埆鏈嶅姟绔繖閲屾槸 websocket-client搴�
+import time
+import websockets
+import asyncio
+from queue import Queue
+# import threading
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--host",
+                    type=str,
+                    default="localhost",
+                    required=False,
+                    help="host ip, localhost, 0.0.0.0")
+parser.add_argument("--port",
+                    type=int,
+                    default=10095,
+                    required=False,
+                    help="grpc server port")
+parser.add_argument("--chunk_size",
+                    type=int,
+                    default=300,
+                    help="ms")
+
+args = parser.parse_args()
+
+voices = Queue()
+async def ws_client():
+    global ws # 瀹氫箟涓�涓叏灞�鍙橀噺ws锛岀敤浜庝繚瀛榳ebsocket杩炴帴瀵硅薄
+    # uri = "ws://11.167.134.197:8899"
+    uri = "ws://{}:{}".format(args.host, args.port)
+    ws = await websockets.connect(uri, subprotocols=["binary"]) # 鍒涘缓涓�涓暱杩炴帴
+    ws.max_size = 1024 * 1024 * 20
+    print("connected ws server")
+    
+async def send(data):
+    global ws # 寮曠敤鍏ㄥ眬鍙橀噺ws
+    try:
+        await ws.send(data) # 閫氳繃ws瀵硅薄鍙戦�佹暟鎹�
+    except Exception as e:
+        print('Exception occurred:', e)
+    
+
+
+asyncio.get_event_loop().run_until_complete(ws_client()) # 鍚姩鍗忕▼
+
+
+# 鍏朵粬鍑芥暟鍙互閫氳繃璋冪敤send(data)鏉ュ彂閫佹暟鎹紝渚嬪锛�
+async def test():
+    #print("2")
+    global voices
+    FORMAT = pyaudio.paInt16
+    CHANNELS = 1
+    RATE = 16000
+    CHUNK = int(RATE / 1000 * args.chunk_size)
+
+    p = pyaudio.PyAudio()
+
+    stream = p.open(format=FORMAT,
+                    channels=CHANNELS,
+                    rate=RATE,
+                    input=True,
+                    frames_per_buffer=CHUNK)
+
+    while True:
+
+        data = stream.read(CHUNK)
+        
+        voices.put(data)
+        #print(voices.qsize())
+        await asyncio.sleep(0.01)
+    
+      
+
+
+
+async def ws_send():
+    global voices
+    print("started to sending data!")
+    while True:
+        while not voices.empty():
+            data = voices.get()
+            voices.task_done()
+            await send(data)
+            await asyncio.sleep(0.01)
+        await asyncio.sleep(0.01)
+
+async def main():
+    task = asyncio.create_task(test()) # 鍒涘缓涓�涓悗鍙颁换鍔�
+    task2 = asyncio.create_task(ws_send()) # 鍒涘缓涓�涓悗鍙颁换鍔�
+     
+    await asyncio.gather(task, task2)
+
+asyncio.run(main())

--
Gitblit v1.9.1