Update TextGen/router.py
Browse files- TextGen/router.py +5 -5
TextGen/router.py
CHANGED
@@ -17,11 +17,11 @@ from langchain_google_genai import (
|
|
17 |
from TextGen import app
|
18 |
from gradio_client import Client, handle_file
|
19 |
from typing import List
|
20 |
-
from elevenlabs.client import
|
21 |
from elevenlabs import stream
|
22 |
|
23 |
|
24 |
-
Eleven_client =
|
25 |
api_key=os.environ["ELEVEN_API_KEY"], # Defaults to ELEVEN_API_KEY
|
26 |
)
|
27 |
|
@@ -167,9 +167,9 @@ async def generate_wav(message: VoiceMessage):
|
|
167 |
except Exception as e:
|
168 |
raise HTTPException(status_code=500, detail=str(e))
|
169 |
@app.post("/generate_voice")
|
170 |
-
|
171 |
-
|
172 |
-
|
173 |
yield chunk
|
174 |
|
175 |
return StreamingResponse(audio_stream(), media_type="audio/wav")
|
|
|
17 |
from TextGen import app
|
18 |
from gradio_client import Client, handle_file
|
19 |
from typing import List
|
20 |
+
from elevenlabs.client import ElevenLabs
|
21 |
from elevenlabs import stream
|
22 |
|
23 |
|
24 |
+
Eleven_client = ElevenLabs(
|
25 |
api_key=os.environ["ELEVEN_API_KEY"], # Defaults to ELEVEN_API_KEY
|
26 |
)
|
27 |
|
|
|
167 |
except Exception as e:
|
168 |
raise HTTPException(status_code=500, detail=str(e))
|
169 |
@app.post("/generate_voice")
|
170 |
+
def generate_voice(message: VoiceMessage):
|
171 |
+
def audio_stream():
|
172 |
+
for chunk in Eleven_client.generate(text=message.input, stream=True):
|
173 |
yield chunk
|
174 |
|
175 |
return StreamingResponse(audio_stream(), media_type="audio/wav")
|