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: