Update TextGen/router.py

TextGen/router.py

@@ -17,6 +17,15 @@ from langchain_google_genai import (
 from TextGen import app
 from gradio_client import Client, handle_file
 from typing import List
+from elevenlabs.client import ElevenLabs
+from elevenlabs import stream
+
+
+Eleven_client = ElevenLabs(
+  api_key=os.environ["ELEVEN_API_KEY"], # Defaults to ELEVEN_API_KEY
+)
+
+
 
 class PlayLastMusic(BaseModel):
     '''plays the lastest created music '''
@@ -159,31 +168,11 @@ async def generate_wav(message: VoiceMessage):
         raise HTTPException(status_code=500, detail=str(e))
 @app.post("/generate_voice")
 async def generate_voice(message: VoiceMessage):
-    try:
-        voice = determine_vocie_from_npc(message.npc, message.genre)
-        audio_file_pth = handle_file(voice)
-
-        # Generator function to yield audio chunks
-        async def audio_stream():
-            result = predict(
-                prompt=message.input,
-                language=message.language,
-                audio_file_pth=audio_file_pth,
-                mic_file_path=None,
-                use_mic=False,
-                voice_cleanup=False,
-                no_lang_auto_detect=False,
-                agree=True,
-            )
-            for sampling_rate, audio_chunk in result:
-                yield audio_chunk.tobytes()
-                await asyncio.sleep(0)  # Yield control to the event loop
-
-        # Return the generated audio as a streaming response
-        return StreamingResponse(audio_stream(), media_type="audio/wav")
-
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    audio_stream = Eleven_client.generate(
+      text=message.input,
+      stream=True
+    )
+    return audio_stream
 
 @app.get("/generate_song")
 async def generate_song(text: str):