From cbe4817bc1b21fe6c3f38210bd38e5d54b464c33 Mon Sep 17 00:00:00 2001
From: huangmingming <huangmingming@deepscience.cn>
Date: 星期一, 30 一月 2023 17:18:20 +0800
Subject: [PATCH] adjust import lib order

---
 funasr/runtime/python/grpc/grpc_main_client_mic.py |   14 +++++---------
 funasr/runtime/python/grpc/grpc_server.py          |    5 ++---
 funasr/runtime/python/grpc/grpc_client.py          |    6 ++----
 funasr/runtime/python/grpc/grpc_main_server.py     |    5 +++--
 funasr/runtime/python/grpc/.gitignore              |    1 +
 5 files changed, 13 insertions(+), 18 deletions(-)

diff --git a/funasr/runtime/python/grpc/.gitignore b/funasr/runtime/python/grpc/.gitignore
new file mode 100644
index 0000000..eeb8a6e
--- /dev/null
+++ b/funasr/runtime/python/grpc/.gitignore
@@ -0,0 +1 @@
+**/__pycache__
diff --git a/funasr/runtime/python/grpc/grpc_client.py b/funasr/runtime/python/grpc/grpc_client.py
index d92632f..8f0bcd9 100644
--- a/funasr/runtime/python/grpc/grpc_client.py
+++ b/funasr/runtime/python/grpc/grpc_client.py
@@ -1,12 +1,10 @@
-import paraformer_pb2
-import paraformer_pb2_grpc
-import grpc
 import queue
+import paraformer_pb2
 
 def transcribe_audio_bytes(stub, chunk, user='zksz', language='zh-CN', speaking = True, isEnd = False):
     req = paraformer_pb2.Request()
     if chunk is not None:
-        req.audio_data = chunk.tostring()
+        req.audio_data = chunk
     req.user = user
     req.language = language
     req.speaking = speaking
diff --git a/funasr/runtime/python/grpc/grpc_main_client_mic.py b/funasr/runtime/python/grpc/grpc_main_client_mic.py
index 8a8fe4d..acbe90b 100644
--- a/funasr/runtime/python/grpc/grpc_main_client_mic.py
+++ b/funasr/runtime/python/grpc/grpc_main_client_mic.py
@@ -1,23 +1,19 @@
 import pyaudio
-import scipy.io.wavfile as wav
-import grpc_client
 import grpc
 import json
-from grpc_client import transcribe_audio_bytes
-from paraformer_pb2_grpc import ASRStub
 import webrtcvad
-import numpy as np
 import time
 import asyncio
-import datetime
 import argparse
+
+from grpc_client import transcribe_audio_bytes
+from paraformer_pb2_grpc import ASRStub
 
 async def deal_chunk(sig_mic):
     global stub,SPEAKING,asr_user,language,sample_rate
-    sig = np.frombuffer(sig_mic, 'int16')
-    if vad.is_speech(sig.tobytes(), sample_rate): #speaking
+    if vad.is_speech(sig_mic, sample_rate): #speaking
         SPEAKING = True
-        response = transcribe_audio_bytes(stub, sig, user=asr_user, language=language, speaking = True, isEnd = False) #speaking, send audio to server.
+        response = transcribe_audio_bytes(stub, sig_mic, user=asr_user, language=language, speaking = True, isEnd = False) #speaking, send audio to server.
     else: #silence   
         begin_time = 0
         if SPEAKING: #means we have some audio recorded, send recognize order to server.
diff --git a/funasr/runtime/python/grpc/grpc_main_server.py b/funasr/runtime/python/grpc/grpc_main_server.py
index 9f7e8c2..f3b2348 100644
--- a/funasr/runtime/python/grpc/grpc_main_server.py
+++ b/funasr/runtime/python/grpc/grpc_main_server.py
@@ -1,8 +1,9 @@
+import grpc
 from concurrent import futures
+import argparse
+
 import paraformer_pb2_grpc
 from grpc_server import ASRServicer
-import grpc
-import argparse
 
 def serve(args):
       server = grpc.server(futures.ThreadPoolExecutor(max_workers=10),
diff --git a/funasr/runtime/python/grpc/grpc_server.py b/funasr/runtime/python/grpc/grpc_server.py
index 2924925..19b7354 100644
--- a/funasr/runtime/python/grpc/grpc_server.py
+++ b/funasr/runtime/python/grpc/grpc_server.py
@@ -1,13 +1,12 @@
 from concurrent import futures
 import grpc
 import json
-import paraformer_pb2
-import paraformer_pb2_grpc
 import time
 
-from paraformer_pb2 import Response
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
+import paraformer_pb2_grpc
+from paraformer_pb2 import Response
 
 
 class ASRServicer(paraformer_pb2_grpc.ASRServicer):

--
Gitblit v1.9.1