Update TextGen/router.py

TextGen/router.py

@@ -1,6 +1,9 @@
 import os
 import time
 from io import BytesIO
+import io
+import wave
+import numpy as np
 from langchain_core.pydantic_v1 import BaseModel, Field
 from fastapi import FastAPI, HTTPException, Query, Request
 from fastapi.responses import StreamingResponse,Response
@@ -285,6 +288,7 @@ async def generate_wav(message: VoiceMessage):
 #            yield chunk
 
  #   return StreamingResponse(audio_stream(), media_type="audio/mpeg")
+
 @app.get("/generate_voice_coqui", response_class=StreamingResponse)
 @app.post("/generate_voice_coqui", response_class=StreamingResponse)
 def generate_voice_coqui(message: VoiceMessage = None):
@@ -306,13 +310,23 @@ def generate_voice_coqui(message: VoiceMessage = None):
                 no_lang_auto_detect=False,
                 agree=True,
             )
-        # Generate the audio stream from ElevenLabs
+        
         for chunk in result:
-            print("received : ",chunk)
-            # Assuming chunk is a tuple, where the second element is the audio data array
-            yield chunk[1].tobytes()
-
-    return StreamingResponse(audio_stream()) 
+            print("received : ", chunk)
+            sample_rate, audio_data = chunk  # Assuming chunk[0] is sample rate, chunk[1] is audio NumPy array
+
+            # Convert NumPy array to bytes in WAV format
+            with io.BytesIO() as wav_buffer:
+                with wave.open(wav_buffer, 'wb') as wav_file:
+                    wav_file.setnchannels(1)  # Mono channel
+                    wav_file.setsampwidth(2)  # Sample width in bytes (2 bytes for int16)
+                    wav_file.setframerate(sample_rate)
+                    wav_file.writeframes(audio_data.astype(np.int16).tobytes())
+                
+                wav_buffer.seek(0)
+                yield wav_buffer.read()
+
+    return StreamingResponse(audio_stream(), media_type="audio/wav")
 @app.post("/generate_song")
 @app.get("/generate_song")
 async def generate_song(request:SongRequest,httprequest: Request):