mirror of
https://github.com/BoardWare-Genius/jarvis-models.git
synced 2025-12-13 16:53:24 +00:00
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import io
|
|
import time
|
|
from ntpath import join
|
|
|
|
from fastapi import Request, Response, status
|
|
from fastapi.responses import JSONResponse
|
|
from .blackbox import Blackbox
|
|
from ..tts.tts_service import TTService
|
|
|
|
class TTS(Blackbox):
|
|
|
|
def __init__(self, *args, **kwargs) -> None:
|
|
self.tts_service = TTService("catmaid")
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
return self.processing(*args, **kwargs)
|
|
|
|
def processing(self, *args, **kwargs) -> io.BytesIO:
|
|
text = args[0]
|
|
current_time = time.time()
|
|
audio = self.tts_service.read(text)
|
|
print("#### TTS Service consume : ", (time.time()-current_time))
|
|
return audio
|
|
|
|
def valid(self, *args, **kwargs) -> bool:
|
|
text = args[0]
|
|
return isinstance(text, str)
|
|
|
|
async def fast_api_handler(self, request: Request) -> Response:
|
|
try:
|
|
data = await request.json()
|
|
except:
|
|
return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
|
|
text = data.get("text")
|
|
if text is None:
|
|
return JSONResponse(content={"error": "text is required"}, status_code=status.HTTP_400_BAD_REQUEST)
|
|
by = self.processing(text)
|
|
return Response(content=by.read(), media_type="audio/wav", headers={"Content-Disposition": "attachment; filename=audio.wav"}) |