ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Blackbox melotts updated

Browse Source
This commit is contained in:
gdw6463
2024-05-17 16:19:07 +08:00
parent 4ffa38592c
commit f2d87ec1ea
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/blackbox/melotts.py
View File

@ -26,12 +26,15 @@ class MeloTTS(Blackbox):
text = args[0]
return isinstance(text, str)
def processing(self, *args, **kwargs) -> io.BytesIO:
def processing(self, *args, **kwargs) -> bytes:
text = args[0]
current_time = time.time()
audio = self.tts_service.read(text)
message = {
"text": text
}
response = requests.post(self.melotts, json=message)
print("#### MeloTTS Service consume : ", (time.time()-current_time))
return audio
return response.content
async def fast_api_handler(self, request: Request) -> Response:
try:
@ -41,7 +44,6 @@ class MeloTTS(Blackbox):
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/mp3", headers={"Content-Disposition": "attachment; filename=audio.mp3"})
return Response(content=self.processing(text), media_type="audio/wav", headers={"Content-Disposition": "attachment; filename=audio.wav"})