diff --git a/src/blackbox/melotts.py b/src/blackbox/melotts.py index 1dd9372..76a07e1 100644 --- a/src/blackbox/melotts.py +++ b/src/blackbox/melotts.py @@ -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"})