[server] move dir

speechserving/speechserving/main.py → paddlespeech/server/bin/main.py

@@ -12,15 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import argparse
-
 import uvicorn
 import yaml
-from engine.engine_factory import EngineFactory
 from fastapi import FastAPI
-from restful.api import setup_router
 
-from utils.config import get_config
-from utils.log import logger
+from paddlespeech.server.engine.engine_factory import EngineFactory
+from paddlespeech.server.restful.api import setup_router
+from paddlespeech.server.utils.config import get_config
+from paddlespeech.server.utils.log import logger
 
 app = FastAPI(
     title="PaddleSpeech Serving API", description="Api", version="0.0.1")

speechserving/speechserving/conf/application.yaml → paddlespeech/server/conf/application.yaml

@@ -13,5 +13,5 @@ port: 8090
 
 engine_backend:
     asr: 'conf/asr/asr.yaml'
-    tts: 'conf/tts/tts.yaml'
+    tts: 'conf/tts/tts_pd.yaml'
 

speechserving/speechserving/engine/asr/python/asr_engine.py → paddlespeech/server/engine/asr/python/asr_engine.py

@@ -20,15 +20,15 @@
 import librosa
 import paddle
 import soundfile
-from engine.base_engine import BaseEngine
 
 from paddlespeech.cli.asr.infer import ASRExecutor
 from paddlespeech.cli.log import logger
 from paddlespeech.s2t.frontend.featurizer.text_featurizer import TextFeaturizer
 from paddlespeech.s2t.transform.transformation import Transformation
 from paddlespeech.s2t.utils.dynamic_import import dynamic_import
 from paddlespeech.s2t.utils.utility import UpdateConfig
-from utils.config import get_config
+from paddlespeech.server.engine.base_engine import BaseEngine
+from paddlespeech.server.utils.config import get_config
 
 __all__ = ['ASREngine']
 

speechserving/speechserving/engine/engine_factory.py → paddlespeech/server/engine/engine_factory.py

@@ -13,8 +13,9 @@
 # limitations under the License.
 from typing import Text
 
-from engine.asr.python.asr_engine import ASREngine
-from engine.tts.python.tts_engine import TTSEngine
+from paddlespeech.server.engine.asr.python.asr_engine import ASREngine
+#from paddlespeech.server.engine.tts.python.tts_engine import TTSEngine
+from paddlespeech.server.engine.tts.paddleinference.tts_engine import TTSEngine
 
 
 __all__ = ['EngineFactory']

speechserving/speechserving/engine/tts/paddleinference/tts_engine.py → paddlespeech/server/engine/tts/paddleinference/tts_engine.py

@@ -20,7 +20,6 @@
 import numpy as np
 import paddle
 import soundfile as sf
-from engine.base_engine import BaseEngine
 from scipy.io import wavfile
 
 from paddlespeech.cli.log import logger
@@ -29,12 +28,13 @@
 from paddlespeech.cli.utils import MODEL_HOME
 from paddlespeech.t2s.frontend import English
 from paddlespeech.t2s.frontend.zh_frontend import Frontend
-from utils.audio_process import change_speed
-from utils.config import get_config
-from utils.errors import ErrorCode
-from utils.exception import ServerBaseException
-from utils.paddle_predictor import init_predictor
-from utils.paddle_predictor import run_model
+from paddlespeech.server.engine.base_engine import BaseEngine
+from paddlespeech.server.utils.audio_process import change_speed
+from paddlespeech.server.utils.config import get_config
+from paddlespeech.server.utils.errors import ErrorCode
+from paddlespeech.server.utils.exception import ServerBaseException
+from paddlespeech.server.utils.paddle_predictor import init_predictor
+from paddlespeech.server.utils.paddle_predictor import run_model
 
 __all__ = ['TTSEngine']
 
@@ -339,7 +339,7 @@ def __init__(self):
         """
         super(TTSEngine, self).__init__()
 
-    def init(self, config_file: str):
+    def init(self, config_file: str) -> bool:
         self.executor = TTSServerExecutor()
         self.config_file = config_file
         self.config = get_config(config_file)
@@ -361,6 +361,7 @@ def init(self, config_file: str):
             voc_predictor_conf=self.config.voc_predictor_conf, )
 
         logger.info("Initialize TTS server engine successfully.")
+        return True
 
     def postprocess(self,
                     wav,

speechserving/speechserving/engine/tts/python/tts_engine.py → paddlespeech/server/engine/tts/python/tts_engine.py

@@ -17,15 +17,15 @@
 import librosa
 import numpy as np
 import soundfile as sf
-from engine.base_engine import BaseEngine
 from scipy.io import wavfile
 
 from paddlespeech.cli.log import logger
 from paddlespeech.cli.tts.infer import TTSExecutor
-from utils.audio_process import change_speed
-from utils.config import get_config
-from utils.errors import ErrorCode
-from utils.exception import ServerBaseException
+from paddlespeech.server.utils.audio_process import change_speed
+from paddlespeech.server.engine.base_engine import BaseEngine
+from paddlespeech.server.utils.config import get_config
+from paddlespeech.server.utils.errors import ErrorCode
+from paddlespeech.server.utils.exception import ServerBaseException
 
 __all__ = ['TTSEngine']
 
@@ -48,7 +48,7 @@ def __init__(self, name=None):
         """
         super(TTSEngine, self).__init__()
 
-    def init(self, config_file: str):
+    def init(self, config_file: str) -> bool:
         self.executor = TTSServerExecutor()
         self.config_file = config_file
         self.config = get_config(config_file)
@@ -68,6 +68,7 @@ def init(self, config_file: str):
             lang=self.config.lang)
 
         logger.info("Initialize TTS server engine successfully.")
+        return True
 
     def postprocess(self,
                     wav,

speechserving/speechserving/restful/api.py → paddlespeech/server/restful/api.py

@@ -15,8 +15,8 @@
 
 from fastapi import APIRouter
 
-from .asr_api import router as asr_router
-from .tts_api import router as tts_router
+from paddlespeech.server.restful.asr_api import router as asr_router
+from paddlespeech.server.restful.tts_api import router as tts_router
 
 _router = APIRouter()
 

speechserving/speechserving/restful/asr_api.py → paddlespeech/server/restful/asr_api.py

@@ -14,16 +14,15 @@
 import base64
 import traceback
 from typing import Union
-
-from engine.asr.python.asr_engine import ASREngine
 from fastapi import APIRouter
 
-from .request import ASRRequest
-from .response import ASRResponse
-from .response import ErrorResponse
-from utils.errors import ErrorCode
-from utils.errors import failed_response
-from utils.exception import ServerBaseException
+from paddlespeech.server.engine.asr.python.asr_engine import ASREngine
+from paddlespeech.server.restful.request import ASRRequest
+from paddlespeech.server.restful.response import ASRResponse
+from paddlespeech.server.restful.response import ErrorResponse
+from paddlespeech.server.utils.errors import ErrorCode
+from paddlespeech.server.utils.errors import failed_response
+from paddlespeech.server.utils.exception import ServerBaseException
 
 router = APIRouter()
 

speechserving/speechserving/restful/tts_api.py → paddlespeech/server/restful/tts_api.py

@@ -13,16 +13,16 @@
 # limitations under the License.
 import traceback
 from typing import Union
-
-from engine.tts.python.tts_engine import TTSEngine
 from fastapi import APIRouter
 
-from .request import TTSRequest
-from .response import ErrorResponse
-from .response import TTSResponse
-from utils.errors import ErrorCode
-from utils.errors import failed_response
-from utils.exception import ServerBaseException
+#from paddlespeech.server.engine.tts.python.tts_engine import TTSEngine
+from paddlespeech.server.engine.tts.paddleinference.tts_engine import TTSEngine
+from paddlespeech.server.restful.request import TTSRequest
+from paddlespeech.server.restful.response import ErrorResponse
+from paddlespeech.server.restful.response import TTSResponse
+from paddlespeech.server.utils.errors import ErrorCode
+from paddlespeech.server.utils.errors import failed_response
+from paddlespeech.server.utils.exception import ServerBaseException
 
 router = APIRouter()
 

speechserving/tests/tts/test_client.py → paddlespeech/server/tests/tts/test_client.py

@@ -23,14 +23,7 @@
 import requests
 import soundfile
 
-
-def wav2pcm(wavfile: str, pcmfile: str, data_type=np.int16):
-    with open(wavfile, "rb") as f:
-        f.seek(0)
-        f.read(44)
-        data = np.fromfile(f, dtype=data_type)
-        data.tofile(pcmfile)
-
+from paddlespeech.server.utils.audio_process import wav2pcm
 
 # Request and response
 def tts_client(args):

speechserving/speechserving/utils/exception.py → paddlespeech/server/utils/exception.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import traceback
 
-from utils.errors import ErrorMsg
+from paddlespeech.server.utils.errors import ErrorMsg
 
 
 class ServerBaseException(Exception):