Update TextGen/router.py

TextGen/router.py

@@ -166,21 +166,14 @@ async def generate_wav(message: VoiceMessage):
 
  except Exception as e:
  raise HTTPException(status_code=500, detail=str(e))
-@app.get("/generate_voice")
-async def generate_voice(message: VoiceMessage):
- audio_stream = Eleven_client.generate(
- text=message.input,
- stream=True
- )
- return StreamingResponse(audio_stream, media_type="audio/wav")
- 
 @app.post("/generate_voice")
 async def generate_voice(message: VoiceMessage):
- audio_stream = Eleven_client.generate(
- text=message.input,
- stream=True
- )
- return StreamingResponse(audio_stream, media_type="audio/wav")
+ async def audio_stream():
+ async for chunk in Eleven_client.generate(text=message.input, stream=True):
+ yield chunk
+
+ return StreamingResponse(audio_stream(), media_type="audio/wav")
+ 
 @app.get("/generate_song")
 async def generate_song(text: str):
  try: