mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
99ecaca869
* add python funasr_api supoort * change little to README.md * add core tools stream * modified a little * fix bug for timeout * support for buffer decode * add ffmpeg decode for buffer
73 lines
1.6 KiB
Python
73 lines
1.6 KiB
Python
"""
|
|
Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights
|
|
Reserved. MIT License (https://opensource.org/licenses/MIT)
|
|
|
|
2023-2024 by zhaomingwork@qq.com
|
|
"""
|
|
|
|
# pip install websocket-client
|
|
# apt install ffmpeg
|
|
|
|
import threading
|
|
import traceback
|
|
import json
|
|
import time
|
|
|
|
|
|
# class for recognizer in websocket
|
|
class FunasrStream:
|
|
"""
|
|
python asr recognizer lib
|
|
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
funasr_core
|
|
|
|
):
|
|
"""
|
|
uri: ws or wss server uri
|
|
msg_callback: for message received
|
|
timeout: timeout for get result
|
|
"""
|
|
try:
|
|
self.funasr_core=funasr_core
|
|
|
|
except Exception as e:
|
|
print("FunasrStream init Exception:", e)
|
|
traceback.print_exc()
|
|
|
|
|
|
# feed data to asr engine in stream way
|
|
def feed_chunk(self, chunk):
|
|
try:
|
|
if self.funasr_core is None:
|
|
print("error in stream, funasr_core is None")
|
|
exit(0)
|
|
self.funasr_core.feed_chunk(chunk)
|
|
return
|
|
except:
|
|
print("feed chunk error")
|
|
return
|
|
|
|
|
|
|
|
# return all result for this stream
|
|
def wait_for_end(self):
|
|
try:
|
|
|
|
message = json.dumps({"is_speaking": False})
|
|
self.funasr_core.websocket.send(message)
|
|
self.funasr_core.wait_for_result()
|
|
self.funasr_core.close()
|
|
|
|
# return the msg
|
|
return self.funasr_core.rec_text
|
|
except Exception as e:
|
|
print("error get_final_result ",e)
|
|
return ""
|
|
|
|
|
|
|